Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.h405 using ReductionGenerator = std ::function<llvm::Value*( variable
412 ReductionGenerator MatchReductionGenerator(HloComputation* function,
418 const ReductionGenerator& reduction_generator,
H A Dir_emitter.cc1460 IrEmitter::ReductionGenerator IrEmitter::MatchReductionGenerator( in MatchReductionGenerator()
1627 const ReductionGenerator& reduction_generator, in EmitInnerLoopForVectorizedReduction()
1744 ReductionGenerator reduction_generator = in EmitVectorizedReduce()