Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/
H A Dlhlo_legalize_to_parallel_loops.cc242 SmallVector<Value, 2> parallelLower, parallelUpper, parallelStep; in createReduceOpInNestedParallelLoops() local
254 (isReducingDim ? reduceStep : parallelStep).push_back(step); in createReduceOpInNestedParallelLoops()
263 parallelUpper, parallelStep); in createReduceOpInNestedParallelLoops()