Searched refs:instruction_can_change_layout_func (Results 1 – 8 of 8) sorted by relevance
34 instruction_can_change_layout_func, in CpuLayoutAssignment() argument37 std::move(instruction_can_change_layout_func)), in CpuLayoutAssignment()
34 instruction_can_change_layout_func, in GpuLayoutAssignment() argument37 std::move(instruction_can_change_layout_func)), in GpuLayoutAssignment()
110 instruction_can_change_layout_func) in HloTestBase() argument114 instruction_can_change_layout_func) {} in HloTestBase()121 instruction_can_change_layout_func) in HloTestBase() argument130 instruction_can_change_layout_func); in HloTestBase()
136 instruction_can_change_layout_func = {});145 instruction_can_change_layout_func = {});
227 instruction_can_change_layout_func = {},233 std::move(instruction_can_change_layout_func)) { in target_metadata_()
304 instruction_can_change_layout_func = InstructionCanChangeLayout,
1329 instruction_can_change_layout_func) in InstructionVerifier() argument1331 instruction_can_change_layout_func) {} in InstructionVerifier()
970 instruction_can_change_layout_func, in LayoutAssignment()977 std::move(instruction_can_change_layout_func)) { in LayoutAssignment()