Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_parallel_loops.cc238 SmallVector<Value, 2> reduce_lower, reduce_upper, reduce_step; in CreateReduceOpInNestedParallelLoops() local
248 (is_reducing_dim ? reduce_upper : parallel_upper).push_back(ub); in CreateReduceOpInNestedParallelLoops()
262 loc, reduce_lower, reduce_upper, reduce_step, ValueRange(init_value)); in CreateReduceOpInNestedParallelLoops()