Home
last modified time | relevance | path

Searched defs:first_reduce (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.h237 const HloInstruction* first_reduce) { in AreFusedReductionOutputsConsistent()
Dir_emission_utils.cc595 const HloInstruction* first_reduce) { in IsFusedReductionOutputConsistent()
614 mlir::mhlo::ReduceOp first_reduce) { in IsFusedReductionOutputConsistent()
Dir_emitter_unnested.cc4222 mlir::Operation* first_reduce = nullptr; in EmitPrologueForReduction() local
5295 mlir::Operation* unnested_hlo, mlir::Operation* first_reduce) { in ComputeReductionCodegenInfo()
5569 mlir::Operation* first_reduce = nullptr; in EmitReductionFromOrToContiguousDimensions() local