Home
last modified time | relevance | path

Searched refs:reduce_inst (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc4224 mlir::Operation* reduce_inst = in EmitPrologueForReduction() local
4227 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()