Home
last modified time | relevance | path

Searched defs:expected_array (Results 1 – 10 of 10) 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.cc757 Array2D<bool> expected_array({{false, false}, {false, true}}); in XLA_TEST_F() local
785 Array2D<int32> expected_array({{0, -6}, {4, 5}}); in XLA_TEST_F() local
813 Array2D<uint32> expected_array({{0, 0}, {3, 0}}); in XLA_TEST_F() local
841 Array2D<bool> expected_array({{false, true}, {true, true}}); in XLA_TEST_F() local
869 Array2D<int32> expected_array({{5, -1}, {12, 9}}); in XLA_TEST_F() local
897 Array2D<uint32> expected_array({{5, 7}, {12, 9}}); in XLA_TEST_F() local
925 Array2D<bool> expected_array({{false, true}, {true, false}}); in XLA_TEST_F() local
953 Array2D<int32> expected_array({{5, 6}, {12, 9}}); in XLA_TEST_F() local
981 Array2D<uint32> expected_array({{5, 6}, {12, 9}}); in XLA_TEST_F() local
1006 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.cc612 Array2D<float> expected_array({ in XLA_TEST_P() local
741 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.cc617 auto expected_array = absl::make_unique<Array4D<float>>(8, 5, 1, 1); in TEST_P() local
662 auto expected_array = absl::make_unique<Array2D<float>>(1, 5); in TEST_P() local
706 auto expected_array = absl::make_unique<Array2D<float>>(0, 9); in TEST_P() local
746 auto expected_array = Array2D<float>({ in TEST_P() local
835 auto expected_array = Array2D<float>({ in TEST_P() local
885 auto expected_array = Array3D<float>({{{expected_1}}, {{expected_2}}}); in TEST_P() local
934 Array3D<float> expected_array = {{{11.f, 18.f, 9.f}}}; in TEST_P() local
989 Array4D<float> expected_array(1, 1, 4, 4); in TEST_P() local
1075 Array4D<float> expected_array({{{{2514, 2685}}}}); in TEST_P() local
1153 Array4D<float> expected_array({{{{2514, 2685}}}}); in TEST_P() local
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_conditional_accumulator_test.py56 def _assertEqual_nparray(self, expected_array, result, sess): argument