Searched refs:ReductionGenerator (Results 1 – 2 of 2) sorted by relevance
384 using ReductionGenerator = std ::function<llvm::Value*( variable391 ReductionGenerator MatchReductionGenerator(HloComputation* function,397 const ReductionGenerator& reduction_generator,
1401 IrEmitter::ReductionGenerator IrEmitter::MatchReductionGenerator( in MatchReductionGenerator()1566 const ReductionGenerator& reduction_generator, in EmitInnerLoopForVectorizedReduction()1672 ReductionGenerator reduction_generator = in EmitVectorizedReduce()