Searched refs:reduction_generator (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 397 const ReductionGenerator& reduction_generator,
|
| D | ir_emitter.cc | 1566 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()
|