Searched refs:reduction_generator (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 398 const ReductionGenerator& reduction_generator,
|
| D | ir_emitter.cc | 1531 const ReductionGenerator& reduction_generator, in EmitInnerLoopForVectorizedReduction() argument
1590 reduction_generator(&b_, current_accumulator_value, addend); in EmitInnerLoopForVectorizedReduction()
1642 ReductionGenerator reduction_generator = in EmitVectorizedReduce() local
1644 if (!reduction_generator) { in EmitVectorizedReduce()
1743 reduction_generator, array_index, vector_type, in EmitVectorizedReduce()
1777 reduction_generator, array_index, vector_type, in EmitVectorizedReduce()
|