| /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 | 757 Array2D<bool> expected_array({{false, false}, {false, true}}); in XLA_TEST_F() local
758 ComputeAndCompareR2<bool>(&builder, expected_array, {}); in XLA_TEST_F()
785 Array2D<int32> expected_array({{0, -6}, {4, 5}}); in XLA_TEST_F() local
786 ComputeAndCompareR2<int32>(&builder, expected_array, {}); in XLA_TEST_F()
813 Array2D<uint32> expected_array({{0, 0}, {3, 0}}); in XLA_TEST_F() local
814 ComputeAndCompareR2<uint32>(&builder, expected_array, {}); in XLA_TEST_F()
841 Array2D<bool> expected_array({{false, true}, {true, true}}); in XLA_TEST_F() local
842 ComputeAndCompareR2<bool>(&builder, expected_array, {}); in XLA_TEST_F()
869 Array2D<int32> expected_array({{5, -1}, {12, 9}}); in XLA_TEST_F() local
870 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 | 612 Array2D<float> expected_array({ in XLA_TEST_P() local
627 Literal expected = LiteralUtil::CreateR2FromArray2D<float>(expected_array); in XLA_TEST_P()
741 Array2D<float> expected_array(5, 60); in XLA_TEST_P() local
743 expected_array(indices[0], indices[2] * 30 + indices[1] * 3 + indices[3]) = in XLA_TEST_P()
746 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/contrib/seq2seq/python/kernel_tests/ |
| D | beam_search_decoder_test.py | 86 expected_array = np.array(
95 expected_array = ops.convert_to_tensor(
96 expected_array, dtype=dtypes.float32)
111 expected_array = array_ops.reshape(
112 expected_array, [max_time, batch_size * beam_width])
116 expected_array = _tile_in_depth(expected_array)
123 expected_array = sess.run(expected_array)
124 self.assertAllEqual(expected_array, sorted_array)
156 expected_array = np.expand_dims(np.array(
170 expected_array = ops.convert_to_tensor(
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 617 auto expected_array = absl::make_unique<Array4D<float>>(8, 5, 1, 1); in TEST_P() local
618 expected_array->Fill(kPadValue); in TEST_P()
619 (*expected_array)(1, 0, 0, 0) = 1.0f; in TEST_P()
620 (*expected_array)(1, 2, 0, 0) = 2.0f; in TEST_P()
621 (*expected_array)(4, 0, 0, 0) = 3.0f; in TEST_P()
622 (*expected_array)(4, 2, 0, 0) = 4.0f; in TEST_P()
623 (*expected_array)(7, 0, 0, 0) = 5.0f; in TEST_P()
624 (*expected_array)(7, 2, 0, 0) = 6.0f; in TEST_P()
626 auto expected = LiteralUtil::CreateR4FromArray4D<float>(*expected_array); in TEST_P()
662 auto expected_array = absl::make_unique<Array2D<float>>(1, 5); in TEST_P() local
[all …]
|
| /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/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)
|