Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dcompilation_cache_test.cc144 auto rowmaj_array = LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
149 auto colmaj_array = LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
Dclient_test.cc59 Literal expected_literal = LiteralUtil::CreateR2WithLayout<int32>( in XLA_TEST_F()
Dparams_test.cc459 Literal literal = LiteralUtil::CreateR2WithLayout<float>( in XLA_TEST_F()
471 Literal literal = LiteralUtil::CreateR2WithLayout<float>( in XLA_TEST_F()
Dmap_test.cc368 Literal param0_literal = LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
373 Literal param1_literal = LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
Dcompute_constant_test.cc259 Literal expected_literal = LiteralUtil::CreateR2WithLayout<int32>( in XLA_TEST_F()
Dtransfer_manager_test.cc155 Literal literal = LiteralUtil::CreateR2WithLayout<float>( in XLA_TEST_F()
Dlocal_client_execute_test.cc131 auto x_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
137 auto y_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
Ddot_operation_test.cc505 ->TransferToServer(LiteralUtil::CreateR2WithLayout<complex64>( in XLA_TEST_F()
510 ->TransferToServer(LiteralUtil::CreateR2WithLayout<complex64>( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cse_test.cc84 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
87 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
116 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
119 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
Dlayout_assignment_test.cc165 auto constant_literal1 = LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
167 auto constant_literal2 = LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
216 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
219 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
253 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
256 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
823 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
Dhlo_evaluator_test.cc541 auto input_literal = LiteralUtil::CreateR2WithLayout<int32>( in TEST_P()
543 auto expected = LiteralUtil::CreateR2WithLayout<float>( in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h80 static Literal CreateR2WithLayout(
331 /* static */ Literal LiteralUtil::CreateR2WithLayout( in CreateR2WithLayout() function
346 return CreateR2WithLayout(values, LayoutUtil::GetDefaultLayoutForR2()); in CreateR2()
Dliteral_test.cc745 auto mat_dim0minor = LiteralUtil::CreateR2WithLayout<int32>( in TEST_F()
756 auto mat_dim0major = LiteralUtil::CreateR2WithLayout<int32>( in TEST_F()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc105 auto array = xla::LiteralUtil::CreateR2WithLayout<float>(v, layout); in FloatMatrix()
909 xla::LiteralUtil::CreateR2WithLayout<float>({{18.0f}, {44.0f}}, layout); in TEST()