Searched defs:instruction_can_change_layout_func (Results 1 – 5 of 5) sorted by relevance
34 instruction_can_change_layout_func, in GpuLayoutAssignment()
34 instruction_can_change_layout_func, in CpuLayoutAssignment()
110 instruction_can_change_layout_func) in HloTestBase()121 instruction_can_change_layout_func) in HloTestBase()
1329 instruction_can_change_layout_func) in InstructionVerifier()
969 std::function<bool(const HloInstruction*)> in LayoutAssignment()