Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h397 const ReductionGenerator& reduction_generator,
Dir_emitter.cc1566 const ReductionGenerator& reduction_generator, in EmitInnerLoopForVectorizedReduction() argument
1625 reduction_generator(&b_, current_accumulator_value, addend); in EmitInnerLoopForVectorizedReduction()
1672 ReductionGenerator reduction_generator = in EmitVectorizedReduce() local
1674 if (!reduction_generator) { in EmitVectorizedReduce()
1763 reduction_generator, array_index, vector_type, in EmitVectorizedReduce()
1797 reduction_generator, array_index, vector_type, in EmitVectorizedReduce()