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.h179 static Literal CreateR4FromArray4D(const Array4D<NativeT>& values);
526 return CreateR4FromArray4D(array); in CreateR4Projected()
530 /* static */ Literal LiteralUtil::CreateR4FromArray4D( in CreateR4FromArray4D() function
Dreference_util.cc481 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(lhs); in ConvArray4DGeneralDimensionsDilated()
482 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.cc66 input_literal_ = LiteralUtil::CreateR4FromArray4D(input_array_); in BatchNormalizationTest()
513 LiteralUtil::CreateR4FromArray4D<float>(normalized); in XLA_TEST_P()
517 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P()
617 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P()
800 LiteralUtil::CreateR4FromArray4D<float>(grad_activation); in XLA_TEST_P()
802 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P()
807 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()
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()
Dconstants_test.cc146 Literal input_literal = LiteralUtil::CreateR4FromArray4D(input_array); in TEST_F()
Ddot_operation_test.cc578 ->TransferToServer(LiteralUtil::CreateR4FromArray4D<T>( in XLA_TYPED_TEST()
586 ->TransferToServer(LiteralUtil::CreateR4FromArray4D<T>( in XLA_TYPED_TEST()
732 ->TransferToServer(LiteralUtil::CreateR4FromArray4D<T>( in XLA_TYPED_TEST()
740 ->TransferToServer(LiteralUtil::CreateR4FromArray4D<T>( in XLA_TYPED_TEST()
Dclient_library_test_base.h526 LiteralUtil::CreateR4FromArray4D<NativeT>(expected); in ComputeAndCompareR4()
543 LiteralUtil::CreateR4FromArray4D<NativeT>(expected); in ComputeAndCompareR4()
/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.cc600 auto input = LiteralUtil::CreateR4FromArray4D<float>(input_array); in TEST_P()
626 auto expected = LiteralUtil::CreateR4FromArray4D<float>(*expected_array); in TEST_P()
851 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*lhs_array); in TEST_P()
857 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*rhs_array); in TEST_P()
952 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(lhs_array); in TEST_P()
963 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(rhs_array); in TEST_P()
998 auto expected = LiteralUtil::CreateR4FromArray4D<float>(expected_array); in TEST_P()
1028 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(input); in TEST_P()
1032 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(weight); in TEST_P()
1078 auto expected = LiteralUtil::CreateR4FromArray4D<float>( in TEST_P()
[all …]
Dhlo_cost_analysis_test.cc505 LiteralUtil::CreateR4FromArray4D(Array4D<float>(2, 3, 4, 5)))); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_rewriter_test.cc611 LiteralUtil::CreateR4FromArray4D(constant_arr).ToStringWithoutShape(); in TEST_F()