Searched refs:sum_broadcast (Results 1 – 1 of 1) sorted by relevance
2747 auto sum_broadcast = CommonPrefixBroadcast(loc, logits, sum, rewriter); in matchAndRewrite() local2748 rewriter.replaceOpWithNewOp<mhlo::DivOp>(op, exp, sum_broadcast); in matchAndRewrite()