Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dinfeed_test.cc76 TestInfeedRoundTrip(LiteralUtil::CreateR3WithLayout( in TEST_F()
81 TestInfeedRoundTrip(LiteralUtil::CreateR3WithLayout( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_infeed_test.cc82 TestInfeedRoundTrip(LiteralUtil::CreateR3WithLayout( in TEST_F()
87 TestInfeedRoundTrip(LiteralUtil::CreateR3WithLayout( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h88 static Literal CreateR3WithLayout(
350 /* static */ Literal LiteralUtil::CreateR3WithLayout( in CreateR3WithLayout() function
378 return CreateR3WithLayout(values, LayoutUtil::GetDefaultLayoutForR3()); in CreateR3()
Dliteral_test.cc677 auto expected = LiteralUtil::CreateR3WithLayout<float>({ in TEST_F()
697 auto expected = LiteralUtil::CreateR3WithLayout<float>({ in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc2965 Literal arg = LiteralUtil::CreateR3WithLayout<float>( in TEST_P()