Searched refs:fusion_root (Results 1 – 3 of 3) sorted by relevance
180 HloInstruction* fusion_root = instruction->fused_expression_root(); in insert_on_outputs() local182 << fusion_root->ToString(); in insert_on_outputs()184 TF_ASSIGN_OR_RETURN(bool instruction_changed, insert_after(fusion_root)); in insert_on_outputs()
634 auto fusion_root = fusion_computation->root_instruction(); in ResolveInconsistentFusions() local635 if (ShapeUtil::Compatible(fusion_root->shape(), hlo->shape())) { in ResolveInconsistentFusions()644 fusion_root, in ResolveInconsistentFusions()
717 auto* fusion_root = in ProfitableToMakeDotOperandColumnMajor() local719 if (fusion_root->opcode() != HloOpcode::kAdd) { in ProfitableToMakeDotOperandColumnMajor()723 for (auto* fusion_root_op : fusion_root->operands()) { in ProfitableToMakeDotOperandColumnMajor()