Home
last modified time | relevance | path

Searched refs:CreateR3WithLayout (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dinfeed_test.cc76 TestInfeedRoundTrip(LiteralUtil::CreateR3WithLayout( in TEST_F()
81 TestInfeedRoundTrip(LiteralUtil::CreateR3WithLayout( in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
H A Dcpu_infeed_test.cc82 TestInfeedRoundTrip(LiteralUtil::CreateR3WithLayout( in TEST_F()
87 TestInfeedRoundTrip(LiteralUtil::CreateR3WithLayout( in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dliteral_util.h93 static Literal CreateR3WithLayout(
324 /* static */ Literal LiteralUtil::CreateR3WithLayout( in CreateR3WithLayout() function
352 return CreateR3WithLayout(values, LayoutUtil::GetDefaultLayoutForR3()); in CreateR3()
H A Dliteral_test.cc680 auto expected = LiteralUtil::CreateR3WithLayout<float>({ in TEST_F()
700 auto expected = LiteralUtil::CreateR3WithLayout<float>({ in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator_test.cc4290 Literal arg = LiteralUtil::CreateR3WithLayout<float>( in TEST_P()