Searched refs:fusion_instr (Results 1 – 2 of 2) sorted by relevance
192 bool HasOnlyRowMajorLayout(const HloInstruction& fusion_instr) { in HasOnlyRowMajorLayout() argument193 CHECK(fusion_instr.opcode() == HloOpcode::kFusion); in HasOnlyRowMajorLayout()194 auto instrs = fusion_instr.fused_instructions_computation()->instructions(); in HasOnlyRowMajorLayout()
478 auto* fusion_instr = xla::Cast<xla::HloFusionInstruction>(instr); in EmitFusionOp() local498 *fusion_instr->fused_instructions_computation(), in EmitFusionOp()