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.cc1716 int64 innermost_dimension_size = in EmitVectorizedReduce() local
1726 if (innermost_dimension_size >= vectorization_factor) { in EmitVectorizedReduce()
1728 int64 end_index = (innermost_dimension_size / vectorization_factor) * in EmitVectorizedReduce()
1764 if (innermost_dimension_size % vectorization_factor) { in EmitVectorizedReduce()
1767 b_.getInt64(innermost_dimension_size - in EmitVectorizedReduce()
1768 (innermost_dimension_size % vectorization_factor)); in EmitVectorizedReduce()
1772 innermost_dimension_size % vectorization_factor); in EmitVectorizedReduce()