Searched refs:ForceParameterLayout (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | custom_call_test.cc | 151 ForceParameterLayout(module.get(), 0, LayoutUtil::MakeLayout({1, 0})); in XLA_TEST_F()
181 ForceParameterLayout(module.get(), 0, LayoutUtil::MakeLayout({1, 0})); in XLA_TEST_F()
|
| D | hlo_test_base.h | 265 void ForceParameterLayout(HloModule* module, int64 param_no, in ForceParameterLayout() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 1337 ForceParameterLayout(m.get(), 0, r2_dim0major); in TEST_F()
1338 ForceParameterLayout(m.get(), 1, r2_dim0major); in TEST_F()
|