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.cc197 Literal literal = LiteralUtil::CreateR2WithLayout<float>( in XLA_TEST_F()
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()
Ddot_operation_test.cc549 ->TransferToServer(LiteralUtil::CreateR2WithLayout<complex64>( in XLA_TEST_F()
554 ->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()
820 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
Dhlo_evaluator_test.cc618 auto input_literal = LiteralUtil::CreateR2WithLayout<int32>( in TEST_P()
620 auto expected = LiteralUtil::CreateR2WithLayout<float>( in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h79 static Literal CreateR2WithLayout(
313 /* static */ Literal LiteralUtil::CreateR2WithLayout( in CreateR2WithLayout() function
328 return CreateR2WithLayout(values, LayoutUtil::GetDefaultLayoutForR2()); in CreateR2()
Dliteral_test.cc782 auto mat_dim0minor = LiteralUtil::CreateR2WithLayout<int32>( in TEST_F()
793 auto mat_dim0major = LiteralUtil::CreateR2WithLayout<int32>( in TEST_F()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc209 auto array = xla::LiteralUtil::CreateR2WithLayout<float>(v, layout); in FloatMatrix()
1614 xla::LiteralUtil::CreateR2WithLayout<float>({{18.0f}, {44.0f}}, layout); in TEST()