Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1735 int64 innermost_dimension = LayoutUtil::Minor(reduce->shape().layout(), 0); in EmitVectorizedReduce() local
1737 reduce->shape().dimensions(innermost_dimension); in EmitVectorizedReduce()
1752 absl::StrFormat("dim.%d", innermost_dimension)); in EmitVectorizedReduce()
1753 array_multi_index[innermost_dimension] = loop->GetIndVarValue(); in EmitVectorizedReduce()
1786 array_multi_index[innermost_dimension] = in EmitVectorizedReduce()