Searched refs:reduce_inst (Results 1 – 1 of 1) sorted by relevance
4224 mlir::Operation* reduce_inst = in EmitPrologueForReduction() local4227 if (!IsReductionFromOrToContiguousDimensions(reduce_inst)) { in EmitPrologueForReduction()4231 auto results = GetHloOutputs(reduce_inst); in EmitPrologueForReduction()4235 VLOG(10) << "Emit prologue for reduction: " << MlirToString(reduce_inst); in EmitPrologueForReduction()4237 first_reduce = reduce_inst; in EmitPrologueForReduction()4241 reduce_inst->getAttrOfType<mlir::DenseIntElementsAttr>( in EmitPrologueForReduction()