Home
last modified time | relevance | path

Searched refs:CreateR2FromArray2D (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util_test.cc55 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result); in TEST_F()
67 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result); in TEST_F()
98 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result); in TEST_F()
118 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result); in TEST_F()
156 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result); in TEST_F()
164 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result); in TEST_F()
434 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*actual); in TEST_F()
H A Dliteral_util.h138 static Literal CreateR2FromArray2D(const Array2D<NativeT>& values);
418 /* static */ Literal LiteralUtil::CreateR2FromArray2D( in CreateR2FromArray2D() function
507 return CreateR2FromArray2D(array); in MakeIdentityR2()
H A Dliteral_util.cc385 return CreateR2FromArray2D(*value); in CreateR2F32Linspace()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtriangular_solve_expander_test.cc79 auto la = LiteralUtil::CreateR2FromArray2D(a); in TEST_P()
80 auto lb = LiteralUtil::CreateR2FromArray2D(b); in TEST_P()
93 auto ref_lb = LiteralUtil::CreateR2FromArray2D(*ref_b); in TEST_P()
H A Dhlo_evaluator_test.cc184 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in ReduceWindowMaxIotaTest()
758 auto input = LiteralUtil::CreateR2FromArray2D<float>(*input_array); in TEST_P()
783 auto expected = LiteralUtil::CreateR2FromArray2D<float>(*expected_array); in TEST_P()
799 auto input = LiteralUtil::CreateR2FromArray2D<float>(*input_array); in TEST_P()
822 auto expected = LiteralUtil::CreateR2FromArray2D<float>(*expected_array); in TEST_P()
839 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); in TEST_P()
868 auto expected = LiteralUtil::CreateR2FromArray2D<float>(expected_array); in TEST_P()
891 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array); in TEST_P()
923 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); in TEST_P()
935 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array); in TEST_P()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dreduce_test.cc181 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR1PredTest()
215 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR0Test()
243 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR1Test()
301 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunVectorizedReduceTestForType()
455 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in XLA_TEST_F()
487 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in XLA_TEST_F()
598 auto input_literal = LiteralUtil::CreateR2FromArray2D(input); in XLA_TEST_F()
613 auto input_literal = LiteralUtil::CreateR2FromArray2D(input); in XLA_TEST_F()
627 auto input_literal = LiteralUtil::CreateR2FromArray2D(input); in XLA_TEST_F()
639 auto input_literal = LiteralUtil::CreateR2FromArray2D(input); in XLA_TEST_F()
[all …]
H A Dreshape_test.cc57 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P()
102 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P()
135 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P()
150 LiteralUtil::CreateR2FromArray2D<float>(Array2D<float>(0, 3)); in XLA_TEST_P()
164 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P()
659 Literal expected = LiteralUtil::CreateR2FromArray2D<float>(expected_array); in XLA_TEST_P()
783 auto expected = LiteralUtil::CreateR2FromArray2D(expected_array); in XLA_TEST_P()
H A Dmatrix_ops_simple_test.cc65 LiteralUtil::CreateR2FromArray2D<T>({{2.71828f, 1.00000f}, // row 0 in XLA_TYPED_TEST()
94 LiteralUtil::CreateR2FromArray2D<T>({{1.5f, 0.5f}, // row 0 in XLA_TYPED_TEST()
113 LiteralUtil::CreateR2FromArray2D<T>({{7.0f, 6.0f}, // row 0 in XLA_TYPED_TEST()
H A Ddot_operation_test.cc206 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
210 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
709 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
736 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
883 LiteralUtil::CreateR2FromArray2D<T>(*arg_0_value_array))); in XLA_TYPED_TEST()
887 LiteralUtil::CreateR2FromArray2D<T>(*arg_1_value_array))); in XLA_TYPED_TEST()
891 LiteralUtil::CreateR2FromArray2D<T>(*arg_2_value_array))); in XLA_TYPED_TEST()
932 LiteralUtil::CreateR2FromArray2D<T>(*arg_0_value_array))); in XLA_TYPED_TEST()
936 LiteralUtil::CreateR2FromArray2D<T>(*arg_1_value_array))); in XLA_TYPED_TEST()
940 LiteralUtil::CreateR2FromArray2D<T>(*arg_2_value_array))); in XLA_TYPED_TEST()
H A Ddynamic_ops_test.cc153 std::move(LiteralUtil::CreateR2FromArray2D(input_values_int) in RunR2()
157 std::move(LiteralUtil::CreateR2FromArray2D(expected_values_int) in RunR2()
437 std::move(LiteralUtil::CreateR2FromArray2D(input_values_int) in RunR2()
441 std::move(LiteralUtil::CreateR2FromArray2D(update_values_int) in RunR2()
445 std::move(LiteralUtil::CreateR2FromArray2D(expected_values_int) in RunR2()
H A Dliteral_test_util.h184 EXPECT_TRUE(Equal(LiteralUtil::CreateR2FromArray2D(expected), actual)); in ExpectR2EqualArray2D()
241 EXPECT_TRUE(Near(LiteralUtil::CreateR2FromArray2D(expected), actual, error)); in ExpectR2NearArray2D()
H A Dcustom_call_test.cc132 HloInstruction::CreateConstant(LiteralUtil::CreateR2FromArray2D(array))); in XLA_TEST_F()
147 HloInstruction::CreateConstant(LiteralUtil::CreateR2FromArray2D( in XLA_TEST_F()
H A Dcpu_gpu_fusion_test.cc96 LiteralUtil::CreateR2FromArray2D(operand_data[i]))); in TestElementwise2D()
127 auto expected = LiteralUtil::CreateR2FromArray2D(answer_data); in TestElementwise2D()
809 Literal l1 = LiteralUtil::CreateR2FromArray2D(arr).Relayout( in XLA_TEST_F()
812 Literal l2 = LiteralUtil::CreateR2FromArray2D(arr).Relayout( in XLA_TEST_F()
H A Dclient_library_test_base.h503 LiteralUtil::CreateR2FromArray2D<NativeT>(expected); in ComputeAndCompareR2()
520 LiteralUtil::CreateR2FromArray2D<NativeT>(expected); in ComputeAndCompareR2()
634 Literal literal = LiteralUtil::CreateR2FromArray2D(array_2d); in CreateR2Parameter()
H A Dround_trip_transfer_test.cc87 RoundTripTest(LiteralUtil::CreateR2FromArray2D<float>(Array2D<float>(10, 0))); in TEST_F()
H A Dbroadcast_simple_test.cc75 auto r2_data = LiteralUtil::CreateR2FromArray2D(*r2_array).Relayout( in MakeR2Data()
613 auto expected = LiteralUtil::CreateR2FromArray2D(expected_array); in XLA_TEST_P()
H A Dbroadcast_test.cc248 LiteralUtil::CreateR2FromArray2D<float>(to_broadcast))); in TEST_F()
H A Dparams_test.cc95 LiteralUtil::CreateR2FromArray2D<float>(Array2D<float>(3, 0)); in XLA_TEST_F()
H A Dconcat_test.cc763 auto input = LiteralUtil::CreateR2FromArray2D<float>(*input_array); in XLA_TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
H A Dcpu_external_constants_test.cc44 LiteralUtil::CreateR2FromArray2D(backing_array))); in TestWithArray()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.h2769 LiteralUtil::CreateR2FromArray2D<NativeT>(values)); in ConstantR2FromArray2D()