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()
116 instruction_can_change_layout_func = {});125 instruction_can_change_layout_func = {});
91 instruction_can_change_layout_func) in HloTestBase() argument95 instruction_can_change_layout_func) {} in HloTestBase()102 instruction_can_change_layout_func) in HloTestBase() argument111 instruction_can_change_layout_func); in HloTestBase()
268 instruction_can_change_layout_func = {},274 std::move(instruction_can_change_layout_func)) { in target_metadata_()
306 instruction_can_change_layout_func = InstructionCanChangeLayout,
1791 instruction_can_change_layout_func) in InstructionVerifier() argument1793 instruction_can_change_layout_func) {} in InstructionVerifier()
1027 instruction_can_change_layout_func, in LayoutAssignment()1034 std::move(instruction_can_change_layout_func)) { in LayoutAssignment()