Home
last modified time | relevance | path

Searched refs:CreateR4FromArray4D (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util_test.cc129 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*result); in TEST_F()
146 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*result); in TEST_F()
202 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*result); in TEST_F()
215 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*result); in TEST_F()
278 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*actual); in TEST_F()
312 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*actual); in TEST_F()
368 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*actual); in TEST_F()
421 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*actual); in TEST_F()
H A Dliteral_util.h148 static Literal CreateR4FromArray4D(const Array4D<NativeT>& values);
485 return CreateR4FromArray4D(array); in CreateR4Projected()
489 /* static */ Literal LiteralUtil::CreateR4FromArray4D( in CreateR4FromArray4D() function
H A Dreference_util.cc418 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(lhs); in ConvArray4DGeneralDimensionsDilated()
419 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(rhs); in ConvArray4DGeneralDimensionsDilated()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbroadcast_test.cc169 LiteralUtil::CreateR4FromArray4D<float>(expected), result, error_spec_)); in TEST_F()
199 LiteralUtil::CreateR4FromArray4D<float>(expected), result, error_spec_)); in TEST_F()
220 EXPECT_TRUE(LiteralTestUtil::Near(LiteralUtil::CreateR4FromArray4D(r4_array), in XLA_TEST_F()
241 LiteralUtil::CreateR4FromArray4D<float>(expected), result, error_spec_)); in TEST_F()
263 LiteralUtil::CreateR4FromArray4D<float>(expected), result, error_spec_)); in TEST_F()
294 LiteralUtil::CreateR4FromArray4D<float>(expected), result, error_spec_)); in TEST_F()
H A Dbatch_normalization_test.cc62 input_literal_ = LiteralUtil::CreateR4FromArray4D(input_array_); in BatchNormalizationTest()
576 LiteralUtil::CreateR4FromArray4D<float>(normalized); in XLA_TEST_P()
580 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P()
680 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P()
863 LiteralUtil::CreateR4FromArray4D<float>(grad_activation); in XLA_TEST_P()
865 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P()
870 LiteralUtil::CreateR4FromArray4D<float>(grad_output_array); in XLA_TEST_P()
H A Dcopy_test.cc82 TestCopyOp(LiteralUtil::CreateR4FromArray4D(Array4D<int32_t>(0, 2, 3, 2))); in XLA_TEST_F()
202 Literal literal = LiteralUtil::CreateR4FromArray4D(a); in TestCopyConstantLayoutR4()
H A Dliteral_test_util.h196 EXPECT_TRUE(Equal(LiteralUtil::CreateR4FromArray4D(expected), actual)); in ExpectR4EqualArray4D()
255 EXPECT_TRUE(Near(LiteralUtil::CreateR4FromArray4D(expected), actual, error)); in ExpectR4NearArray4D()
H A Ddot_operation_test.cc623 ->TransferToServer(LiteralUtil::CreateR4FromArray4D<T>( in XLA_TYPED_TEST()
631 ->TransferToServer(LiteralUtil::CreateR4FromArray4D<T>( in XLA_TYPED_TEST()
773 ->TransferToServer(LiteralUtil::CreateR4FromArray4D<T>( in XLA_TYPED_TEST()
781 ->TransferToServer(LiteralUtil::CreateR4FromArray4D<T>( in XLA_TYPED_TEST()
1732 AddParam(LiteralUtil::CreateR4FromArray4D<float>(input_arr), &builder); in XLA_TEST_F()
1773 AddParam(LiteralUtil::CreateR4FromArray4D<float>(input_arr), &builder); in XLA_TEST_F()
H A Dclient_library_test_base.h557 LiteralUtil::CreateR4FromArray4D<NativeT>(expected); in ComputeAndCompareR4()
574 LiteralUtil::CreateR4FromArray4D<NativeT>(expected); in ComputeAndCompareR4()
660 Literal literal = LiteralUtil::CreateR4FromArray4D(array_4d); in CreateR4Parameter()
H A Dpad_test.cc208 auto input = LiteralUtil::CreateR4FromArray4D<float>(input_array); in TEST_P()
254 auto input = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P()
H A Dround_trip_transfer_test.cc137 RoundTripTest(LiteralUtil::CreateR4FromArray4D<float>(array4d)); in TEST_F()
H A Dconvolution_dimension_numbers_test.cc93 client_->TransferToServer(LiteralUtil::CreateR4FromArray4D(*weight_array)) in XLA_TEST_F()
H A Dconstants_test.cc146 Literal input_literal = LiteralUtil::CreateR4FromArray4D(input_array); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dconv_canonicalization_test.cc63 LiteralUtil::CreateR4FromArray4D(Array4D<float>( in TEST_F()
67 LiteralUtil::CreateR4FromArray4D(Array4D<float>( in TEST_F()
126 LiteralUtil::CreateR4FromArray4D(Array4D<float>( in TEST_F()
130 LiteralUtil::CreateR4FromArray4D(Array4D<float>( in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator_test.cc715 auto input = LiteralUtil::CreateR4FromArray4D<float>(input_array); in TEST_P()
741 auto expected = LiteralUtil::CreateR4FromArray4D<float>(*expected_array); in TEST_P()
966 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*lhs_array); in TEST_P()
972 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*rhs_array); in TEST_P()
1067 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(lhs_array); in TEST_P()
1078 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(rhs_array); in TEST_P()
1113 auto expected = LiteralUtil::CreateR4FromArray4D<float>(expected_array); in TEST_P()
1143 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(input); in TEST_P()
1147 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(weight); in TEST_P()
1193 auto expected = LiteralUtil::CreateR4FromArray4D<float>( in TEST_P()
[all …]
H A Dhlo_cost_analysis_test.cc833 LiteralUtil::CreateR4FromArray4D(Array4D<float>(2, 3, 4, 5)))); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dinfeed_test.cc97 TestInfeedRoundTrip(LiteralUtil::CreateR4FromArray4D<float>(array)); in TEST_F()
115 {LiteralUtil::CreateR4FromArray4D<float>(array), in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_rewriter_test.cc622 LiteralUtil::CreateR4FromArray4D(constant_arr).ToStringWithoutShape(); in TEST_F()