| /external/tensorflow/tensorflow/python/debug/cli/ |
| cli_test_utils.py | 48 def assert_array_lines_close(test, expected_array, array_lines):
55 expected_array: Expected value of the array.
65 test.assertAllClose(np.array(expected_array).flatten(), elements)
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| vector_ops_reduce_test.cc | 139 Array2D<float> expected_array({{6.0f, 15.0f}, {6.0f, 15.0f}, {6.0f, 15.0f}});
141 ComputeAndCompareR2<float>(&builder_, expected_array, {}, errspec_);
150 Array2D<float> expected_array(
153 ComputeAndCompareR2<float>(&builder_, expected_array, {}, errspec_);
162 Array2D<float> expected_array({{3.0f, 6.0f, 9.0f}, {12.0f, 15.0f, 18.0f}});
164 ComputeAndCompareR2<float>(&builder_, expected_array, {}, errspec_);
|
| pad_test.cc | 216 Array4D<float> expected_array(1, 1, 5, 8);
217 expected_array.Fill(pad_value);
218 expected_array(0, 0, 2, 2) = 1.0f;
219 expected_array(0, 0, 2, 3) = 2.0f;
220 expected_array(0, 0, 2, 4) = 3.0f;
221 expected_array(0, 0, 3, 2) = 4.0f;
222 expected_array(0, 0, 3, 3) = 5.0f;
223 expected_array(0, 0, 3, 4) = 6.0f;
224 ComputeAndCompareR4<float>(&b, expected_array, {}, ErrorSpec(0.0001));
262 Array4D<float> expected_array(1, 25, 17, 11)
[all...] |
| array_elementwise_ops_test.cc | 757 Array2D<bool> expected_array({{false, false}, {false, true}});
758 ComputeAndCompareR2<bool>(&builder, expected_array, {});
785 Array2D<int32> expected_array({{0, -6}, {4, 5}});
786 ComputeAndCompareR2<int32>(&builder, expected_array, {});
813 Array2D<uint32> expected_array({{0, 0}, {3, 0}});
814 ComputeAndCompareR2<uint32>(&builder, expected_array, {});
841 Array2D<bool> expected_array({{false, true}, {true, true}});
842 ComputeAndCompareR2<bool>(&builder, expected_array, {});
869 Array2D<int32> expected_array({{5, -1}, {12, 9}});
870 ComputeAndCompareR2<int32>(&builder, expected_array, {});
[all...] |
| broadcast_simple_test.cc | 348 Array3D<float> expected_array(spec.output_bounds[0], spec.output_bounds[1],
358 int n1 = expected_array.n1();
359 int n2 = expected_array.n2();
360 int n3 = expected_array.n3();
364 Each({i, j, k}, &expected_array(i, j, k));
368 auto expected = LiteralUtil::CreateR3FromArray3D(expected_array);
603 Array2D<float> expected_array(spec.output_bounds[0], spec.output_bounds[1]);
605 expected_array.Each([&](int64 i, int64 j, float* v) {
615 auto expected = LiteralUtil::CreateR2FromArray2D(expected_array);
|
| reshape_test.cc | 612 Array2D<float> expected_array({
627 Literal expected = LiteralUtil::CreateR2FromArray2D<float>(expected_array);
741 Array2D<float> expected_array(5, 60);
743 expected_array(indices[0], indices[2] * 30 + indices[1] * 3 + indices[3]) =
746 auto expected = LiteralUtil::CreateR2FromArray2D(expected_array);
[all...] |
| params_test.cc | 115 Array2D<float> expected_array(
117 ComputeAndCompareR2<float>(&builder, expected_array, {param0_data.get()},
|
| map_test.cc | 298 Array2D<float> expected_array(
300 ComputeAndCompareR2<float>(&builder, expected_array, {param0_data.get()},
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| 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
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_evaluator_test.cc | 617 auto expected_array = absl::make_unique<Array4D<float>>(8, 5, 1, 1); local
618 expected_array->Fill(kPadValue);
619 (*expected_array)(1, 0, 0, 0) = 1.0f;
620 (*expected_array)(1, 2, 0, 0) = 2.0f;
621 (*expected_array)(4, 0, 0, 0) = 3.0f;
622 (*expected_array)(4, 2, 0, 0) = 4.0f;
623 (*expected_array)(7, 0, 0, 0) = 5.0f;
624 (*expected_array)(7, 2, 0, 0) = 6.0f;
626 auto expected = LiteralUtil::CreateR4FromArray4D<float>(*expected_array);
662 auto expected_array = absl::make_unique<Array2D<float>>(1, 5) local
706 auto expected_array = absl::make_unique<Array2D<float>>(0, 9); local
746 auto expected_array = Array2D<float>({ local
835 auto expected_array = Array2D<float>({ local
885 auto expected_array = Array3D<float>({{{expected_1}}, {{expected_2}}}); local
934 Array3D<float> expected_array = {{{11.f, 18.f, 9.f}}}; local
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| sparse_conditional_accumulator_test.py | 56 def _assertEqual_nparray(self, expected_array, result, sess):
57 expected_tensor = _indexedslice(expected_array)
|
