Searched refs:fusion_root (Results 1 – 2 of 2) sorted by relevance
752 auto fusion_root = fusion_computation->root_instruction(); in ResolveInconsistentFusions() local753 if (ShapeUtil::Compatible(fusion_root->shape(), hlo->shape())) { in ResolveInconsistentFusions()762 fusion_root, in ResolveInconsistentFusions()
904 auto* fusion_root = in ProfitableToMakeDotOperandColumnMajor() local906 if (fusion_root->opcode() != HloOpcode::kAdd) { in ProfitableToMakeDotOperandColumnMajor()910 for (auto* fusion_root_op : fusion_root->operands()) { in ProfitableToMakeDotOperandColumnMajor()