Home
last modified time | relevance | path

Searched defs:expected_array (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dvector_ops_reduce_test.cc139 Array2D<float> expected_array({{6.0f, 15.0f}, {6.0f, 15.0f}, {6.0f, 15.0f}}); in TEST_F() local
150 Array2D<float> expected_array( in TEST_F() local
162 Array2D<float> expected_array({{3.0f, 6.0f, 9.0f}, {12.0f, 15.0f, 18.0f}}); in TEST_F() local
Darray_elementwise_ops_test.cc812 Array2D<bool> expected_array({{false, false}, {false, true}}); in XLA_TEST_F() local
840 Array2D<int32> expected_array({{0, -6}, {4, 5}}); in XLA_TEST_F() local
868 Array2D<uint32> expected_array({{0, 0}, {3, 0}}); in XLA_TEST_F() local
896 Array2D<bool> expected_array({{false, true}, {true, true}}); in XLA_TEST_F() local
924 Array2D<int32> expected_array({{5, -1}, {12, 9}}); in XLA_TEST_F() local
952 Array2D<uint32> expected_array({{5, 7}, {12, 9}}); in XLA_TEST_F() local
980 Array2D<bool> expected_array({{false, true}, {true, false}}); in XLA_TEST_F() local
1008 Array2D<int32> expected_array({{5, 6}, {12, 9}}); in XLA_TEST_F() local
1036 Array2D<uint32> expected_array({{5, 6}, {12, 9}}); in XLA_TEST_F() local
1061 Array2D<bool> expected_array({{true, false}, {false, true}}); in XLA_TEST_F() local
[all …]
Dpad_test.cc216 Array4D<float> expected_array(1, 1, 5, 8); in TEST_P() local
262 Array4D<float> expected_array(1, 25, 17, 11); in XLA_TEST_P() local
Dbroadcast_simple_test.cc348 Array3D<float> expected_array(spec.output_bounds[0], spec.output_bounds[1], in XLA_TEST_P() local
603 Array2D<float> expected_array(spec.output_bounds[0], spec.output_bounds[1]); in XLA_TEST_P() local
Dparams_test.cc115 Array2D<float> expected_array( in XLA_TEST_F() local
Dreshape_test.cc645 Array2D<float> expected_array({ in XLA_TEST_P() local
779 Array2D<float> expected_array(5, 60); in XLA_TEST_P() local
Dmap_test.cc298 Array2D<float> expected_array( in TEST_F() local
/external/tensorflow/tensorflow/python/debug/cli/
Dcli_test_utils.py48 def assert_array_lines_close(test, expected_array, array_lines): argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc694 auto expected_array = absl::make_unique<Array4D<float>>(8, 5, 1, 1); in TEST_P() local
739 auto expected_array = absl::make_unique<Array2D<float>>(1, 5); in TEST_P() local
783 auto expected_array = absl::make_unique<Array2D<float>>(0, 9); in TEST_P() local
823 auto expected_array = Array2D<float>({ in TEST_P() local
912 auto expected_array = Array2D<float>({ in TEST_P() local
962 auto expected_array = Array3D<float>({{{expected_1}}, {{expected_2}}}); in TEST_P() local
1011 Array3D<float> expected_array = {{{11.f, 18.f, 9.f}}}; in TEST_P() local
1066 Array4D<float> expected_array(1, 1, 4, 4); in TEST_P() local
1152 Array4D<float> expected_array({{{{2514, 2685}}}}); in TEST_P() local
1230 Array4D<float> expected_array({{{{2514, 2685}}}}); in TEST_P() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsorting_test.cc127 Array2D<float> expected_array(2, 1000); in XLA_TEST_F() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_conditional_accumulator_test.py56 def _assertEqual_nparray(self, expected_array, result, sess): argument