Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util_test.cc120 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*result); in TEST_F()
137 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*result); in TEST_F()
193 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*result); in TEST_F()
206 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*result); in TEST_F()
269 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*actual); in TEST_F()
303 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*actual); in TEST_F()
359 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*actual); in TEST_F()
412 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*actual); in TEST_F()
Dliteral_util.h142 static Literal CreateR4FromArray4D(const Array4D<NativeT>& values);
493 return CreateR4FromArray4D(array); in CreateR4Projected()
497 /* static */ Literal LiteralUtil::CreateR4FromArray4D( in CreateR4FromArray4D() function
Dreference_util.cc415 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(lhs); in ConvArray4DGeneralDimensionsDilated()
416 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(rhs); in ConvArray4DGeneralDimensionsDilated()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
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()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_test.cc170 LiteralUtil::CreateR4FromArray4D<float>(expected), result, error_spec_)); in TEST_F()
200 LiteralUtil::CreateR4FromArray4D<float>(expected), result, error_spec_)); in TEST_F()
221 EXPECT_TRUE(LiteralTestUtil::Near(LiteralUtil::CreateR4FromArray4D(r4_array), in XLA_TEST_F()
242 LiteralUtil::CreateR4FromArray4D<float>(expected), result, error_spec_)); in TEST_F()
264 LiteralUtil::CreateR4FromArray4D<float>(expected), result, error_spec_)); in TEST_F()
295 LiteralUtil::CreateR4FromArray4D<float>(expected), result, error_spec_)); in TEST_F()
Dbatch_normalization_test.cc67 input_literal_ = LiteralUtil::CreateR4FromArray4D(input_array_); in BatchNormalizationTest()
611 LiteralUtil::CreateR4FromArray4D<float>(normalized); in XLA_TEST_P()
615 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P()
715 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P()
898 LiteralUtil::CreateR4FromArray4D<float>(grad_activation); in XLA_TEST_P()
900 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P()
905 LiteralUtil::CreateR4FromArray4D<float>(grad_output_array); in XLA_TEST_P()
Dliteral_test_util.h197 EXPECT_TRUE(Equal(LiteralUtil::CreateR4FromArray4D(expected), actual)); in ExpectR4EqualArray4D()
256 EXPECT_TRUE(Near(LiteralUtil::CreateR4FromArray4D(expected), actual, error)); in ExpectR4NearArray4D()
Dcopy_test.cc84 TestCopyOp(LiteralUtil::CreateR4FromArray4D(Array4D<int32>(0, 2, 3, 2))); in XLA_TEST_F()
204 Literal literal = LiteralUtil::CreateR4FromArray4D(a); in TestCopyConstantLayoutR4()
Dround_trip_transfer_test.cc138 RoundTripTest(LiteralUtil::CreateR4FromArray4D<float>(array4d)); in TEST_F()
Dconvolution_dimension_numbers_test.cc96 client_->TransferToServer(LiteralUtil::CreateR4FromArray4D(*weight_array)) in XLA_TEST_F()
Dconstants_test.cc146 Literal input_literal = LiteralUtil::CreateR4FromArray4D(input_array); in TEST_F()
Dpad_test.cc210 auto input = LiteralUtil::CreateR4FromArray4D<float>(input_array); in TEST_P()
256 auto input = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P()
Ddot_operation_test.cc622 ->TransferToServer(LiteralUtil::CreateR4FromArray4D<T>( in XLA_TYPED_TEST()
630 ->TransferToServer(LiteralUtil::CreateR4FromArray4D<T>( in XLA_TYPED_TEST()
772 ->TransferToServer(LiteralUtil::CreateR4FromArray4D<T>( in XLA_TYPED_TEST()
780 ->TransferToServer(LiteralUtil::CreateR4FromArray4D<T>( in XLA_TYPED_TEST()
1731 AddParam(LiteralUtil::CreateR4FromArray4D<float>(input_arr), &builder); in XLA_TEST_F()
1772 AddParam(LiteralUtil::CreateR4FromArray4D<float>(input_arr), &builder); in XLA_TEST_F()
Dclient_library_test_base.h557 LiteralUtil::CreateR4FromArray4D<NativeT>(expected); in ComputeAndCompareR4()
574 LiteralUtil::CreateR4FromArray4D<NativeT>(expected); in ComputeAndCompareR4()
664 Literal literal = LiteralUtil::CreateR4FromArray4D(array_4d); in CreateR4Parameter()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dinfeed_test.cc97 TestInfeedRoundTrip(LiteralUtil::CreateR4FromArray4D<float>(array)); in TEST_F()
115 {LiteralUtil::CreateR4FromArray4D<float>(array), in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc677 auto input = LiteralUtil::CreateR4FromArray4D<float>(input_array); in TEST_P()
703 auto expected = LiteralUtil::CreateR4FromArray4D<float>(*expected_array); in TEST_P()
928 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*lhs_array); in TEST_P()
934 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*rhs_array); in TEST_P()
1029 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(lhs_array); in TEST_P()
1040 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(rhs_array); in TEST_P()
1075 auto expected = LiteralUtil::CreateR4FromArray4D<float>(expected_array); in TEST_P()
1105 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(input); in TEST_P()
1109 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(weight); in TEST_P()
1155 auto expected = LiteralUtil::CreateR4FromArray4D<float>( in TEST_P()
[all …]
Dhlo_cost_analysis_test.cc730 LiteralUtil::CreateR4FromArray4D(Array4D<float>(2, 3, 4, 5)))); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_rewriter_test.cc621 LiteralUtil::CreateR4FromArray4D(constant_arr).ToStringWithoutShape(); in TEST_F()