| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | convolution_test_1d.cc | 219 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
222 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
254 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in RunTest()
257 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in RunTest()
289 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
292 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
320 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
323 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
356 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in RunTest()
359 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in RunTest()
|
| H A D | dynamic_ops_test.cc | 188 std::move(LiteralUtil::CreateR3FromArray3D(input_values_int) in RunR3()
192 std::move(LiteralUtil::CreateR3FromArray3D(expected_values_int) in RunR3()
476 std::move(LiteralUtil::CreateR3FromArray3D(input_values_int) in RunR3()
480 std::move(LiteralUtil::CreateR3FromArray3D(update_values_int) in RunR3()
484 std::move(LiteralUtil::CreateR3FromArray3D(expected_values_int) in RunR3()
563 Literal literal = LiteralUtil::CreateR3FromArray3D<NativeT>(values); in DumpArray()
|
| H A D | dot_operation_test.cc | 669 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>( in XLA_TYPED_TEST()
675 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>( in XLA_TYPED_TEST()
704 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>( in XLA_TYPED_TEST()
742 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>( in XLA_TYPED_TEST()
1694 AddParam(LiteralUtil::CreateR3FromArray3D<float>(input_arr), &builder); in XLA_TEST_F()
1711 AddParam(LiteralUtil::CreateR3FromArray3D<float>(input_arr), &builder); in XLA_TEST_F()
1749 AddParam(LiteralUtil::CreateR3FromArray3D<float>(input_arr), &builder); in XLA_TEST_F()
1844 auto input_literal = LiteralUtil::CreateR3FromArray3D<float>(input_arr); in DOT_ReorderContracting()
|
| H A D | constants_test.cc | 116 ConstantLiteral(&builder, LiteralUtil::CreateR3FromArray3D<float>( in TEST_F()
132 ConstantLiteral(&builder, LiteralUtil::CreateR3FromArray3D<float>(array3d)); in TEST_F()
|
| H A D | literal_test_util.h | 190 EXPECT_TRUE(Equal(LiteralUtil::CreateR3FromArray3D(expected), actual)); in ExpectR3EqualArray3D()
248 EXPECT_TRUE(Near(LiteralUtil::CreateR3FromArray3D(expected), actual, error)); in ExpectR3NearArray3D()
|
| H A D | client_library_test_base.h | 530 LiteralUtil::CreateR3FromArray3D<NativeT>(expected); in ComputeAndCompareR3()
547 LiteralUtil::CreateR3FromArray3D<NativeT>(expected); in ComputeAndCompareR3()
647 Literal literal = LiteralUtil::CreateR3FromArray3D(array_3d); in CreateR3Parameter()
|
| H A D | map_test.cc | 396 LiteralUtil::CreateR3FromArray3D<int32_t>(Array3D<int32_t>(3, 0, 2)); in XLA_TEST_F()
401 LiteralUtil::CreateR3FromArray3D<int32_t>(Array3D<int32_t>(3, 0, 2)); in XLA_TEST_F()
|
| H A D | broadcast_simple_test.cc | 62 auto r3_data = LiteralUtil::CreateR3FromArray3D(*r3_array).Relayout( in MakeR3Data()
366 auto expected = LiteralUtil::CreateR3FromArray3D(expected_array); in XLA_TEST_P()
|
| H A D | copy_test.cc | 168 Literal literal = LiteralUtil::CreateR3FromArray3D(a); in TestCopyConstantLayout021()
|
| H A D | batch_normalization_test.cc | 345 {LiteralUtil::CreateR3FromArray3D<float>(Array3D<float>(260, 2, 2, 1.0f)), in XLA_TEST_F()
377 {LiteralUtil::CreateR3FromArray3D<float>( in XLA_TEST_F()
|
| H A D | broadcast_test.cc | 282 LiteralUtil::CreateR3FromArray3D<float>(input_vals))); in TEST_F()
|
| H A D | reduce_test.cc | 535 Literal input_literal = LiteralUtil::CreateR3FromArray3D(input_data); in XLA_TEST_F()
826 auto input_literal = LiteralUtil::CreateR3FromArray3D(input_array); in XLA_TEST_P()
|
| H A D | concat_test.cc | 859 auto x_literal = LiteralUtil::CreateR3FromArray3D<float>(x3d); in XLA_TEST_F()
|
| H A D | array_elementwise_ops_test.cc | 2325 LiteralUtil::CreateR3FromArray3D<float>(Array3D<float>(0, 7, 0)); in XLA_TEST_F()
2330 LiteralUtil::CreateR3FromArray3D<float>(Array3D<float>(0, 7, 0)); in XLA_TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | reference_util_test.cc | 108 auto actual_literal = LiteralUtil::CreateR3FromArray3D(*result); in TEST_F()
176 auto actual_literal = LiteralUtil::CreateR3FromArray3D(*result); in TEST_F()
189 auto actual_literal = LiteralUtil::CreateR3FromArray3D(*result); in TEST_F()
230 auto actual_literal = LiteralUtil::CreateR3FromArray3D(*actual); in TEST_F()
243 auto actual_literal = LiteralUtil::CreateR3FromArray3D(*actual); in TEST_F()
|
| H A D | literal_util.h | 143 static Literal CreateR3FromArray3D(const Array3D<NativeT>& values);
430 /* static */ Literal LiteralUtil::CreateR3FromArray3D( in CreateR3FromArray3D() function
457 return CreateR3FromArray3D(array); in CreateR3Projected()
|
| H A D | literal_test.cc | 255 auto literal = LiteralUtil::CreateR3FromArray3D(array_3d); in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| H A D | infeed_test.cc | 124 auto literal = LiteralUtil::CreateR3FromArray3D<float>(array); in TEST_F()
168 auto literal = LiteralUtil::CreateR3FromArray3D<float>(array); in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_evaluator_test.cc | 1001 auto expected = LiteralUtil::CreateR3FromArray3D<float>(expected_array); in TEST_P()
1010 auto lhs_literal = LiteralUtil::CreateR3FromArray3D<float>(lhs_array); in TEST_P()
1015 auto rhs_literal = LiteralUtil::CreateR3FromArray3D<float>(rhs_array); in TEST_P()
1050 auto expected = LiteralUtil::CreateR3FromArray3D<float>(expected_array); in TEST_P()
|