| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | pad_test.cc | 216 Array4D<float> expected_array(1, 1, 5, 8); in TEST_P() local
217 expected_array.Fill(pad_value); in TEST_P()
218 expected_array(0, 0, 2, 2) = 1.0f; in TEST_P()
219 expected_array(0, 0, 2, 3) = 2.0f; in TEST_P()
220 expected_array(0, 0, 2, 4) = 3.0f; in TEST_P()
221 expected_array(0, 0, 3, 2) = 4.0f; in TEST_P()
222 expected_array(0, 0, 3, 3) = 5.0f; in TEST_P()
223 expected_array(0, 0, 3, 4) = 6.0f; in TEST_P()
224 ComputeAndCompareR4<float>(&b, expected_array, {}, ErrorSpec(0.0001)); in TEST_P()
262 Array4D<float> expected_array(1, 25, 17, 11); in XLA_TEST_P() local
[all …]
|
| D | vector_ops_reduce_test.cc | 139 Array2D<float> expected_array({{6.0f, 15.0f}, {6.0f, 15.0f}, {6.0f, 15.0f}}); in TEST_F() local
141 ComputeAndCompareR2<float>(&builder_, expected_array, {}, errspec_); in TEST_F()
150 Array2D<float> expected_array( in TEST_F() local
153 ComputeAndCompareR2<float>(&builder_, expected_array, {}, errspec_); in TEST_F()
162 Array2D<float> expected_array({{3.0f, 6.0f, 9.0f}, {12.0f, 15.0f, 18.0f}}); in TEST_F() local
164 ComputeAndCompareR2<float>(&builder_, expected_array, {}, errspec_); in TEST_F()
|
| D | array_elementwise_ops_test.cc | 812 Array2D<bool> expected_array({{false, false}, {false, true}}); in XLA_TEST_F() local
813 ComputeAndCompareR2<bool>(&builder, expected_array, {}); in XLA_TEST_F()
840 Array2D<int32> expected_array({{0, -6}, {4, 5}}); in XLA_TEST_F() local
841 ComputeAndCompareR2<int32>(&builder, expected_array, {}); in XLA_TEST_F()
868 Array2D<uint32> expected_array({{0, 0}, {3, 0}}); in XLA_TEST_F() local
869 ComputeAndCompareR2<uint32>(&builder, expected_array, {}); in XLA_TEST_F()
896 Array2D<bool> expected_array({{false, true}, {true, true}}); in XLA_TEST_F() local
897 ComputeAndCompareR2<bool>(&builder, expected_array, {}); in XLA_TEST_F()
924 Array2D<int32> expected_array({{5, -1}, {12, 9}}); in XLA_TEST_F() local
925 ComputeAndCompareR2<int32>(&builder, expected_array, {}); in XLA_TEST_F()
[all …]
|
| D | broadcast_simple_test.cc | 348 Array3D<float> expected_array(spec.output_bounds[0], spec.output_bounds[1], in XLA_TEST_P() local
358 int n1 = expected_array.n1(); in XLA_TEST_P()
359 int n2 = expected_array.n2(); in XLA_TEST_P()
360 int n3 = expected_array.n3(); in XLA_TEST_P()
364 Each({i, j, k}, &expected_array(i, j, k)); in XLA_TEST_P()
368 auto expected = LiteralUtil::CreateR3FromArray3D(expected_array); in XLA_TEST_P()
603 Array2D<float> expected_array(spec.output_bounds[0], spec.output_bounds[1]); in XLA_TEST_P() local
605 expected_array.Each([&](int64 i, int64 j, float* v) { in XLA_TEST_P()
615 auto expected = LiteralUtil::CreateR2FromArray2D(expected_array); in XLA_TEST_P()
|
| D | reshape_test.cc | 645 Array2D<float> expected_array({ in XLA_TEST_P() local
660 Literal expected = LiteralUtil::CreateR2FromArray2D<float>(expected_array); in XLA_TEST_P()
779 Array2D<float> expected_array(5, 60); in XLA_TEST_P() local
781 expected_array(indices[0], indices[2] * 30 + indices[1] * 3 + indices[3]) = in XLA_TEST_P()
784 auto expected = LiteralUtil::CreateR2FromArray2D(expected_array); in XLA_TEST_P()
|
| D | params_test.cc | 115 Array2D<float> expected_array( in XLA_TEST_F() local
117 ComputeAndCompareR2<float>(&builder, expected_array, {param0_data.get()}, in XLA_TEST_F()
|
| D | map_test.cc | 298 Array2D<float> expected_array( in TEST_F() local
300 ComputeAndCompareR2<float>(&builder, expected_array, {param0_data.get()}, in TEST_F()
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| D | cli_test_utils.py | 48 def assert_array_lines_close(test, expected_array, array_lines): argument
65 test.assertAllClose(np.array(expected_array).flatten(), elements)
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting_test.cc | 127 Array2D<float> expected_array(2, 1000); in XLA_TEST_F() local
128 expected_array.Fill(2.0f); in XLA_TEST_F()
131 ComputeAndCompareR2<float>(&builder, expected_array, {}, error_spec); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 694 auto expected_array = absl::make_unique<Array4D<float>>(8, 5, 1, 1); in TEST_P() local
695 expected_array->Fill(kPadValue); in TEST_P()
696 (*expected_array)(1, 0, 0, 0) = 1.0f; in TEST_P()
697 (*expected_array)(1, 2, 0, 0) = 2.0f; in TEST_P()
698 (*expected_array)(4, 0, 0, 0) = 3.0f; in TEST_P()
699 (*expected_array)(4, 2, 0, 0) = 4.0f; in TEST_P()
700 (*expected_array)(7, 0, 0, 0) = 5.0f; in TEST_P()
701 (*expected_array)(7, 2, 0, 0) = 6.0f; in TEST_P()
703 auto expected = LiteralUtil::CreateR4FromArray4D<float>(*expected_array); in TEST_P()
739 auto expected_array = absl::make_unique<Array2D<float>>(1, 5); in TEST_P() local
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_conditional_accumulator_test.py | 56 def _assertEqual_nparray(self, expected_array, result, sess): argument
57 expected_tensor = _indexedslice(expected_array)
|