Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
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()
Dhlo_evaluator_test.cc159 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in ReduceWindowMaxIotaTest()
720 auto input = LiteralUtil::CreateR2FromArray2D<float>(*input_array); in TEST_P()
745 auto expected = LiteralUtil::CreateR2FromArray2D<float>(*expected_array); in TEST_P()
761 auto input = LiteralUtil::CreateR2FromArray2D<float>(*input_array); in TEST_P()
784 auto expected = LiteralUtil::CreateR2FromArray2D<float>(*expected_array); in TEST_P()
801 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); in TEST_P()
830 auto expected = LiteralUtil::CreateR2FromArray2D<float>(expected_array); in TEST_P()
853 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array); in TEST_P()
885 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); in TEST_P()
897 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array); in TEST_P()
[all …]
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util_test.cc56 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result); in TEST_F()
68 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result); in TEST_F()
99 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result); in TEST_F()
109 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result); in TEST_F()
147 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result); in TEST_F()
155 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result); in TEST_F()
425 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*actual); in TEST_F()
Dliteral_util.h132 static Literal CreateR2FromArray2D(const Array2D<NativeT>& values);
426 /* static */ Literal LiteralUtil::CreateR2FromArray2D( in CreateR2FromArray2D() function
515 return CreateR2FromArray2D(array); in MakeIdentityR2()
Dliteral_util.cc341 return CreateR2FromArray2D(*value); in CreateR2F32Linspace()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmatrix_ops_simple_test.cc67 LiteralUtil::CreateR2FromArray2D<T>({{2.71828f, 1.00000f}, // row 0 in XLA_TYPED_TEST()
96 LiteralUtil::CreateR2FromArray2D<T>({{1.5f, 0.5f}, // row 0 in XLA_TYPED_TEST()
115 LiteralUtil::CreateR2FromArray2D<T>({{7.0f, 6.0f}, // row 0 in XLA_TYPED_TEST()
Dreduce_test.cc180 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR1PredTest()
213 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR0Test()
240 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR1Test()
297 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunVectorizedReduceTestForType()
451 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in XLA_TEST_F()
482 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in XLA_TEST_F()
591 auto input_literal = LiteralUtil::CreateR2FromArray2D(input); in XLA_TEST_F()
606 auto input_literal = LiteralUtil::CreateR2FromArray2D(input); in XLA_TEST_F()
620 auto input_literal = LiteralUtil::CreateR2FromArray2D(input); in XLA_TEST_F()
632 auto input_literal = LiteralUtil::CreateR2FromArray2D(input); in XLA_TEST_F()
[all …]
Dcustom_call_test.cc107 HloInstruction::CreateConstant(LiteralUtil::CreateR2FromArray2D(array))); in XLA_TEST_F()
122 HloInstruction::CreateConstant(LiteralUtil::CreateR2FromArray2D( in XLA_TEST_F()
Dreshape_test.cc58 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P()
103 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P()
136 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P()
151 LiteralUtil::CreateR2FromArray2D<float>(Array2D<float>(0, 3)); in XLA_TEST_P()
165 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P()
660 Literal expected = LiteralUtil::CreateR2FromArray2D<float>(expected_array); in XLA_TEST_P()
784 auto expected = LiteralUtil::CreateR2FromArray2D(expected_array); in XLA_TEST_P()
Ddynamic_ops_test.cc154 std::move(LiteralUtil::CreateR2FromArray2D(input_values_int) in RunR2()
158 std::move(LiteralUtil::CreateR2FromArray2D(expected_values_int) in RunR2()
438 std::move(LiteralUtil::CreateR2FromArray2D(input_values_int) in RunR2()
442 std::move(LiteralUtil::CreateR2FromArray2D(update_values_int) in RunR2()
446 std::move(LiteralUtil::CreateR2FromArray2D(expected_values_int) in RunR2()
Ddot_operation_test.cc207 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
211 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
708 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
735 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
882 LiteralUtil::CreateR2FromArray2D<T>(*arg_0_value_array))); in XLA_TYPED_TEST()
886 LiteralUtil::CreateR2FromArray2D<T>(*arg_1_value_array))); in XLA_TYPED_TEST()
890 LiteralUtil::CreateR2FromArray2D<T>(*arg_2_value_array))); in XLA_TYPED_TEST()
931 LiteralUtil::CreateR2FromArray2D<T>(*arg_0_value_array))); in XLA_TYPED_TEST()
935 LiteralUtil::CreateR2FromArray2D<T>(*arg_1_value_array))); in XLA_TYPED_TEST()
939 LiteralUtil::CreateR2FromArray2D<T>(*arg_2_value_array))); in XLA_TYPED_TEST()
Dliteral_test_util.h185 EXPECT_TRUE(Equal(LiteralUtil::CreateR2FromArray2D(expected), actual)); in ExpectR2EqualArray2D()
242 EXPECT_TRUE(Near(LiteralUtil::CreateR2FromArray2D(expected), actual, error)); in ExpectR2NearArray2D()
Dround_trip_transfer_test.cc88 RoundTripTest(LiteralUtil::CreateR2FromArray2D<float>(Array2D<float>(10, 0))); in TEST_F()
Dcpu_gpu_fusion_test.cc98 LiteralUtil::CreateR2FromArray2D(operand_data[i]))); in TestElementwise2D()
129 auto expected = LiteralUtil::CreateR2FromArray2D(answer_data); in TestElementwise2D()
811 Literal l1 = LiteralUtil::CreateR2FromArray2D(arr).Relayout( in XLA_TEST_F()
814 Literal l2 = LiteralUtil::CreateR2FromArray2D(arr).Relayout( in XLA_TEST_F()
Dclient_library_test_base.h503 LiteralUtil::CreateR2FromArray2D<NativeT>(expected); in ComputeAndCompareR2()
520 LiteralUtil::CreateR2FromArray2D<NativeT>(expected); in ComputeAndCompareR2()
636 Literal literal = LiteralUtil::CreateR2FromArray2D(array_2d); in CreateR2Parameter()
Dbroadcast_simple_test.cc77 auto r2_data = LiteralUtil::CreateR2FromArray2D(*r2_array).Relayout( in MakeR2Data()
615 auto expected = LiteralUtil::CreateR2FromArray2D(expected_array); in XLA_TEST_P()
Dbroadcast_test.cc249 LiteralUtil::CreateR2FromArray2D<float>(to_broadcast))); in TEST_F()
Dparams_test.cc96 LiteralUtil::CreateR2FromArray2D<float>(Array2D<float>(3, 0)); in XLA_TEST_F()
Dconcat_test.cc763 auto input = LiteralUtil::CreateR2FromArray2D<float>(*input_array); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_external_constants_test.cc43 LiteralUtil::CreateR2FromArray2D(backing_array))); in TestWithArray()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h2611 LiteralUtil::CreateR2FromArray2D<NativeT>(values)); in ConstantR2FromArray2D()