def test_prepare(_get_data_shapes): a_shape, b_shape, c_shape, d_shape, out_shape = _get_data_shapes model = _get_simple_model(a_shape, b_shape, c_shape, d_shape, out_shape) backend = NgraphBackend.prepare(model) for idx in range(10): input_a, input_b, input_c, input_d = _get_input_data(a_shape, b_shape, c_shape, d_shape) ng_results = backend.run([input_a, input_b, input_c, input_d]) expected = np.dot(np.abs(input_a + input_b), input_c) + input_d assert np.allclose(ng_results, [expected])
def test_run_model(_get_data_shapes): a_shape, b_shape, c_shape, d_shape, out_shape = _get_data_shapes input_a, input_b, input_c, input_d = _get_input_data(a_shape, b_shape, c_shape, d_shape) model = _get_simple_model(a_shape, b_shape, c_shape, d_shape, out_shape) ng_results = NgraphBackend.run_model(model, [input_a, input_b, input_c, input_d]) expected = np.dot(np.abs(input_a + input_b), input_c) + input_d assert np.allclose(ng_results, [expected])