Searched refs:ForceResultLayout (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | custom_call_test.cc | 152 ForceResultLayout(module.get(), LayoutUtil::MakeLayout({0, 1})); in XLA_TEST_F()
182 ForceResultLayout(module.get(), LayoutUtil::MakeLayout({0, 1})); in XLA_TEST_F()
|
| D | hlo_test_base.h | 276 void ForceResultLayout(HloModule* module, const Layout& layout) { in ForceResultLayout() function
282 void ForceResultLayout(HloModule* module, const Layout& layout, in ForceResultLayout() function
|
| D | copy_test.cc | 182 ForceResultLayout(module.get(), LayoutUtil::MakeLayout({1, 2, 0})); in TestCopyConstantLayout021()
216 ForceResultLayout(module.get(), LayoutUtil::MakeLayout(permutation)); in TestCopyConstantLayoutR4()
|