Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emission_utils.h131 const HloInstruction* first_reduce) { in AreFusedReductionOutputsConsistent()
H A Dir_emission_utils.cc459 const HloInstruction* first_reduce) { in IsFusedReductionOutputConsistent()
H A Dir_emitter_unnested.cc4570 mlir::lmhlo::FusionOp fusion, mlir::mhlo::ReduceOp first_reduce) { in ComputeReductionCodegenInfo()
4921 mlir::mhlo::ReduceOp first_reduce = mlir::cast<mlir::mhlo::ReduceOp>( in EmitUnnestedReduction() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorIteratorReduce.cpp54 auto first_reduce = TensorIterator::reduce_op(buffer_0, iter.input(0)); in two_pass_reduction() local