Home
last modified time | relevance | path

Searched refs:ForceResultLayout (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dcustom_call_test.cc152 ForceResultLayout(module.get(), LayoutUtil::MakeLayout({0, 1})); in XLA_TEST_F()
182 ForceResultLayout(module.get(), LayoutUtil::MakeLayout({0, 1})); in XLA_TEST_F()
Dhlo_test_base.h276 void ForceResultLayout(HloModule* module, const Layout& layout) { in ForceResultLayout() function
282 void ForceResultLayout(HloModule* module, const Layout& layout, in ForceResultLayout() function
Dcopy_test.cc182 ForceResultLayout(module.get(), LayoutUtil::MakeLayout({1, 2, 0})); in TestCopyConstantLayout021()
216 ForceResultLayout(module.get(), LayoutUtil::MakeLayout(permutation)); in TestCopyConstantLayoutR4()