Home
last modified time | relevance | path

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

/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.h169 static Literal CreateR2FromArray2D(const Array2D<NativeT>& values);
459 /* static */ Literal LiteralUtil::CreateR2FromArray2D( in CreateR2FromArray2D() function
548 return CreateR2FromArray2D(array); in MakeIdentityR2()
Dliteral_util.cc289 return CreateR2FromArray2D(*value); in CreateR2F32Linspace()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_test.cc179 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR1PredTest()
212 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR0Test()
239 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR1Test()
296 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunVectorizedReduceTestForType()
450 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in XLA_TEST_F()
481 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in XLA_TEST_F()
590 auto input_literal = LiteralUtil::CreateR2FromArray2D(input); in XLA_TEST_F()
605 auto input_literal = LiteralUtil::CreateR2FromArray2D(input); in XLA_TEST_F()
619 auto input_literal = LiteralUtil::CreateR2FromArray2D(input); in XLA_TEST_F()
631 auto input_literal = LiteralUtil::CreateR2FromArray2D(input); in XLA_TEST_F()
[all …]
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()
Ddot_operation_test.cc200 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
204 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
666 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
694 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
842 LiteralUtil::CreateR2FromArray2D<T>(*arg_0_value_array))); in XLA_TYPED_TEST()
846 LiteralUtil::CreateR2FromArray2D<T>(*arg_1_value_array))); in XLA_TYPED_TEST()
850 LiteralUtil::CreateR2FromArray2D<T>(*arg_2_value_array))); in XLA_TYPED_TEST()
891 LiteralUtil::CreateR2FromArray2D<T>(*arg_0_value_array))); in XLA_TYPED_TEST()
895 LiteralUtil::CreateR2FromArray2D<T>(*arg_1_value_array))); in XLA_TYPED_TEST()
899 LiteralUtil::CreateR2FromArray2D<T>(*arg_2_value_array))); in XLA_TYPED_TEST()
Dreshape_test.cc58 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P()
100 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P()
131 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P()
145 LiteralUtil::CreateR2FromArray2D<float>(Array2D<float>(0, 3)); in XLA_TEST_P()
158 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P()
627 Literal expected = LiteralUtil::CreateR2FromArray2D<float>(expected_array); in XLA_TEST_P()
746 auto expected = LiteralUtil::CreateR2FromArray2D(expected_array); in XLA_TEST_P()
Dcustom_call_test.cc107 HloInstruction::CreateConstant(LiteralUtil::CreateR2FromArray2D(array))); in XLA_TEST_F()
122 HloInstruction::CreateConstant(LiteralUtil::CreateR2FromArray2D( in XLA_TEST_F()
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()
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()
Dclient_library_test_base.h472 LiteralUtil::CreateR2FromArray2D<NativeT>(expected); in ComputeAndCompareR2()
489 LiteralUtil::CreateR2FromArray2D<NativeT>(expected); in ComputeAndCompareR2()
580 Literal literal = LiteralUtil::CreateR2FromArray2D(array_2d); in CreateR2Parameter()
Dfusion_test.cc99 LiteralUtil::CreateR2FromArray2D(operand_data[i]))); in TestElementwise2D()
130 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()
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()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc643 auto input = LiteralUtil::CreateR2FromArray2D<float>(*input_array); in TEST_P()
668 auto expected = LiteralUtil::CreateR2FromArray2D<float>(*expected_array); in TEST_P()
684 auto input = LiteralUtil::CreateR2FromArray2D<float>(*input_array); in TEST_P()
707 auto expected = LiteralUtil::CreateR2FromArray2D<float>(*expected_array); in TEST_P()
724 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); in TEST_P()
753 auto expected = LiteralUtil::CreateR2FromArray2D<float>(expected_array); in TEST_P()
776 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array); in TEST_P()
808 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); in TEST_P()
820 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array); in TEST_P()
841 auto expected = LiteralUtil::CreateR2FromArray2D<float>(expected_array); in TEST_P()
[all …]
/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.h1979 LiteralUtil::CreateR2FromArray2D<NativeT>(values)); in ConstantR2FromArray2D()