Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1736 int64 innermost_dimension_size = in EmitVectorizedReduce() local
1746 if (innermost_dimension_size >= vectorization_factor) { in EmitVectorizedReduce()
1748 int64 end_index = (innermost_dimension_size / vectorization_factor) * in EmitVectorizedReduce()
1784 if (innermost_dimension_size % vectorization_factor) { in EmitVectorizedReduce()
1787 b_.getInt64(innermost_dimension_size - in EmitVectorizedReduce()
1788 (innermost_dimension_size % vectorization_factor)); in EmitVectorizedReduce()
1792 innermost_dimension_size % vectorization_factor); in EmitVectorizedReduce()