Home
last modified time | relevance | path

Searched refs:CreateR2WithLayout (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dcompilation_cache_test.cc141 auto rowmaj_array = LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
145 auto colmaj_array = LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
H A Dparams_test.cc465 Literal literal = LiteralUtil::CreateR2WithLayout<float>( in XLA_TEST_F()
476 Literal literal = LiteralUtil::CreateR2WithLayout<float>( in XLA_TEST_F()
H A Dmap_test.cc369 Literal param0_literal = LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
374 Literal param1_literal = LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
H A Dclient_test.cc58 Literal expected_literal = LiteralUtil::CreateR2WithLayout<int32_t>( in XLA_TEST_F()
H A Dcompute_constant_test.cc259 Literal expected_literal = LiteralUtil::CreateR2WithLayout<int32_t>( in XLA_TEST_F()
H A Dlocal_client_execute_test.cc132 auto x_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
138 auto y_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
H A Dtransfer_manager_test.cc197 Literal literal = LiteralUtil::CreateR2WithLayout<float>( in XLA_TEST_F()
H A Ddot_operation_test.cc548 ->TransferToServer(LiteralUtil::CreateR2WithLayout<complex64>( in XLA_TEST_F()
553 ->TransferToServer(LiteralUtil::CreateR2WithLayout<complex64>( in XLA_TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dliteral_util.h85 static Literal CreateR2WithLayout(
305 /* static */ Literal LiteralUtil::CreateR2WithLayout( in CreateR2WithLayout() function
320 return CreateR2WithLayout(values, LayoutUtil::GetDefaultLayoutForR2()); in CreateR2()
H A Dliteral_test.cc789 auto mat_dim0minor = LiteralUtil::CreateR2WithLayout<int32_t>( in TEST_F()
800 auto mat_dim0major = LiteralUtil::CreateR2WithLayout<int32_t>( in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_assignment_test.cc166 auto constant_literal1 = LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
168 auto constant_literal2 = LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
217 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
220 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
813 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
H A Dhlo_cse_test.cc85 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
88 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
H A Dhlo_evaluator_test.cc656 auto input_literal = LiteralUtil::CreateR2WithLayout<int32_t>( in TEST_P()
658 auto expected = LiteralUtil::CreateR2WithLayout<float>( in TEST_P()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/tests/
H A Draw_api_test.cc209 auto array = xla::LiteralUtil::CreateR2WithLayout<float>(v, layout); in FloatMatrix()
1621 xla::LiteralUtil::CreateR2WithLayout<float>({{18.0f}, {44.0f}}, layout); in TEST()