Home
last modified time | relevance | path

Searched refs:expected_array (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dpad_test.cc214 Array4D<float> expected_array(1, 1, 5, 8); in TEST_P() local
215 expected_array.Fill(pad_value); in TEST_P()
216 expected_array(0, 0, 2, 2) = 1.0f; in TEST_P()
217 expected_array(0, 0, 2, 3) = 2.0f; in TEST_P()
218 expected_array(0, 0, 2, 4) = 3.0f; in TEST_P()
219 expected_array(0, 0, 3, 2) = 4.0f; in TEST_P()
220 expected_array(0, 0, 3, 3) = 5.0f; in TEST_P()
221 expected_array(0, 0, 3, 4) = 6.0f; in TEST_P()
222 ComputeAndCompareR4<float>(&b, expected_array, {}, ErrorSpec(0.0001)); in TEST_P()
260 Array4D<float> expected_array(1, 25, 17, 11); in XLA_TEST_P() local
[all …]
H A Darray_elementwise_ops_test.cc819 Array2D<bool> expected_array({{false, false}, {false, true}}); in XLA_TEST_F() local
820 ComputeAndCompareR2<bool>(&builder, expected_array, {}); in XLA_TEST_F()
847 Array2D<int32_t> expected_array({{0, -6}, {4, 5}}); in XLA_TEST_F() local
848 ComputeAndCompareR2<int32_t>(&builder, expected_array, {}); in XLA_TEST_F()
875 Array2D<uint32_t> expected_array({{0, 0}, {3, 0}}); in XLA_TEST_F() local
876 ComputeAndCompareR2<uint32_t>(&builder, expected_array, {}); in XLA_TEST_F()
903 Array2D<bool> expected_array({{false, true}, {true, true}}); in XLA_TEST_F() local
904 ComputeAndCompareR2<bool>(&builder, expected_array, {}); in XLA_TEST_F()
931 Array2D<int32_t> expected_array({{5, -1}, {12, 9}}); in XLA_TEST_F() local
932 ComputeAndCompareR2<int32_t>(&builder, expected_array, {}); in XLA_TEST_F()
[all …]
H A 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
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()
H A Dbroadcast_simple_test.cc346 Array3D<float> expected_array(spec.output_bounds[0], spec.output_bounds[1], in XLA_TEST_P() local
356 int n1 = expected_array.n1(); in XLA_TEST_P()
357 int n2 = expected_array.n2(); in XLA_TEST_P()
358 int n3 = expected_array.n3(); in XLA_TEST_P()
362 Each({i, j, k}, &expected_array(i, j, k)); in XLA_TEST_P()
366 auto expected = LiteralUtil::CreateR3FromArray3D(expected_array); in XLA_TEST_P()
601 Array2D<float> expected_array(spec.output_bounds[0], spec.output_bounds[1]); in XLA_TEST_P() local
603 expected_array.Each([&](int64_t i, int64_t j, float* v) { in XLA_TEST_P()
613 auto expected = LiteralUtil::CreateR2FromArray2D(expected_array); in XLA_TEST_P()
H A Dreshape_test.cc644 Array2D<float> expected_array({ in XLA_TEST_P() local
659 Literal expected = LiteralUtil::CreateR2FromArray2D<float>(expected_array); in XLA_TEST_P()
778 Array2D<float> expected_array(5, 60); in XLA_TEST_P() local
780 expected_array(indices[0], indices[2] * 30 + indices[1] * 3 + indices[3]) = in XLA_TEST_P()
783 auto expected = LiteralUtil::CreateR2FromArray2D(expected_array); in XLA_TEST_P()
H A Dparams_test.cc114 Array2D<float> expected_array( in XLA_TEST_F() local
116 ComputeAndCompareR2<float>(&builder, expected_array, {param0_data.get()}, in XLA_TEST_F()
H A Dmap_test.cc299 Array2D<float> expected_array( in TEST_F() local
301 ComputeAndCompareR2<float>(&builder, expected_array, {param0_data.get()}, in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator_test.cc732 auto expected_array = std::make_unique<Array4D<float>>(8, 5, 1, 1); in TEST_P() local
733 expected_array->Fill(kPadValue); in TEST_P()
734 (*expected_array)(1, 0, 0, 0) = 1.0f; in TEST_P()
735 (*expected_array)(1, 2, 0, 0) = 2.0f; in TEST_P()
736 (*expected_array)(4, 0, 0, 0) = 3.0f; in TEST_P()
737 (*expected_array)(4, 2, 0, 0) = 4.0f; in TEST_P()
738 (*expected_array)(7, 0, 0, 0) = 5.0f; in TEST_P()
739 (*expected_array)(7, 2, 0, 0) = 6.0f; in TEST_P()
741 auto expected = LiteralUtil::CreateR4FromArray4D<float>(*expected_array); in TEST_P()
777 auto expected_array = std::make_unique<Array2D<float>>(1, 5); in TEST_P() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dsorting_test.cc127 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()
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/cli/
H A Dcli_test_utils.py44 def assert_array_lines_close(test, expected_array, array_lines): argument
61 test.assertAllClose(np.array(expected_array).flatten(), elements)
/aosp_15_r20/packages/services/Car/packages/ScriptExecutor/tests/functional/src/com/android/car/scriptexecutortest/functional/
DScriptExecutorFunctionalTest.java332 long[] expected_array = new long[] {1, 100, 3}; in invokeScript_modifiesArray() local
338 assertThat(mListener.mInterimResult.getLongArray("long_array")).isEqualTo(expected_array); in invokeScript_modifiesArray()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
H A Dsparse_conditional_accumulator_test.py53 def _assertEqual_nparray(self, expected_array, result, sess): argument
54 expected_tensor = _indexedslice(expected_array)