HomeSort by relevance Sort by last modified time
    Searched refs:ForceResultLayout (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
custom_call_test.cc 152 ForceResultLayout(module.get(), LayoutUtil::MakeLayout({0, 1}));
182 ForceResultLayout(module.get(), LayoutUtil::MakeLayout({0, 1}));
hlo_test_base.h 275 void ForceResultLayout(HloModule* module, const Layout& layout) {
281 void ForceResultLayout(HloModule* module, const Layout& layout,
copy_test.cc 182 ForceResultLayout(module.get(), LayoutUtil::MakeLayout({1, 2, 0}));
216 ForceResultLayout(module.get(), LayoutUtil::MakeLayout(permutation));

Completed in 122 milliseconds