Searched refs:ReductionGenerator (Results 1 – 2 of 2) sorted by relevance
385 using ReductionGenerator = std ::function<llvm::Value*( variable392 ReductionGenerator MatchReductionGenerator(HloComputation* function,398 const ReductionGenerator& reduction_generator,
1366 IrEmitter::ReductionGenerator IrEmitter::MatchReductionGenerator( in MatchReductionGenerator()1531 const ReductionGenerator& reduction_generator, in EmitInnerLoopForVectorizedReduction()1642 ReductionGenerator reduction_generator = in EmitVectorizedReduce()