Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1682 const int vectorization_factor = in EmitVectorizedReduce() local
1746 if (innermost_dimension_size >= vectorization_factor) { in EmitVectorizedReduce()
1748 int64 end_index = (innermost_dimension_size / vectorization_factor) * in EmitVectorizedReduce()
1749 vectorization_factor; in EmitVectorizedReduce()
1751 loop_nest.AddLoop(start_index, end_index, vectorization_factor, in EmitVectorizedReduce()
1758 reduce->shape().element_type(), vectorization_factor); in EmitVectorizedReduce()
1784 if (innermost_dimension_size % vectorization_factor) { in EmitVectorizedReduce()
1788 (innermost_dimension_size % vectorization_factor)); in EmitVectorizedReduce()
1792 innermost_dimension_size % vectorization_factor); in EmitVectorizedReduce()