Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_parallel_loops.cc230 DenseSet<int> reducing_dims; in CreateReduceOpInNestedParallelLoops() local
232 reducing_dims.insert(rdim.getSExtValue()); in CreateReduceOpInNestedParallelLoops()
241 const bool is_reducing_dim = reducing_dims.count(dim.index()); in CreateReduceOpInNestedParallelLoops()
285 indices.push_back(reducing_dims.count(i) ? *inner_ivs_it++ in CreateReduceOpInNestedParallelLoops()