/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
H A D | reference_util_test.cc | 55 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 D | literal_util.h | 138 static Literal CreateR2FromArray2D(const Array2D<NativeT>& values); 418 /* static */ Literal LiteralUtil::CreateR2FromArray2D( in CreateR2FromArray2D() function 507 return CreateR2FromArray2D(array); in MakeIdentityR2()
|
H A D | literal_util.cc | 385 return CreateR2FromArray2D(*value); in CreateR2F32Linspace()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | triangular_solve_expander_test.cc | 79 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 D | hlo_evaluator_test.cc | 184 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 D | reduce_test.cc | 181 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 D | reshape_test.cc | 57 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 D | matrix_ops_simple_test.cc | 65 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 D | dot_operation_test.cc | 206 ->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 D | dynamic_ops_test.cc | 153 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 D | literal_test_util.h | 184 EXPECT_TRUE(Equal(LiteralUtil::CreateR2FromArray2D(expected), actual)); in ExpectR2EqualArray2D() 241 EXPECT_TRUE(Near(LiteralUtil::CreateR2FromArray2D(expected), actual, error)); in ExpectR2NearArray2D()
|
H A D | custom_call_test.cc | 132 HloInstruction::CreateConstant(LiteralUtil::CreateR2FromArray2D(array))); in XLA_TEST_F() 147 HloInstruction::CreateConstant(LiteralUtil::CreateR2FromArray2D( in XLA_TEST_F()
|
H A D | cpu_gpu_fusion_test.cc | 96 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 D | client_library_test_base.h | 503 LiteralUtil::CreateR2FromArray2D<NativeT>(expected); in ComputeAndCompareR2() 520 LiteralUtil::CreateR2FromArray2D<NativeT>(expected); in ComputeAndCompareR2() 634 Literal literal = LiteralUtil::CreateR2FromArray2D(array_2d); in CreateR2Parameter()
|
H A D | round_trip_transfer_test.cc | 87 RoundTripTest(LiteralUtil::CreateR2FromArray2D<float>(Array2D<float>(10, 0))); in TEST_F()
|
H A D | broadcast_simple_test.cc | 75 auto r2_data = LiteralUtil::CreateR2FromArray2D(*r2_array).Relayout( in MakeR2Data() 613 auto expected = LiteralUtil::CreateR2FromArray2D(expected_array); in XLA_TEST_P()
|
H A D | broadcast_test.cc | 248 LiteralUtil::CreateR2FromArray2D<float>(to_broadcast))); in TEST_F()
|
H A D | params_test.cc | 95 LiteralUtil::CreateR2FromArray2D<float>(Array2D<float>(3, 0)); in XLA_TEST_F()
|
H A D | concat_test.cc | 763 auto input = LiteralUtil::CreateR2FromArray2D<float>(*input_array); in XLA_TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
H A D | cpu_external_constants_test.cc | 44 LiteralUtil::CreateR2FromArray2D(backing_array))); in TestWithArray()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
H A D | xla_builder.h | 2769 LiteralUtil::CreateR2FromArray2D<NativeT>(values)); in ConstantR2FromArray2D()
|