Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_parallel_loops.cc93 MappedIvs MapWindowIvsToInput(OpTy op, ValueRange ivs, ValueRange window_ivs, in MapWindowIvsToInput() argument
122 Value offset = b->create<SubIOp>(loc, window_ivs[i], pad_low_val); in MapWindowIvsToInput()
532 SmallVector<Value, 2> window_ivs; member
569 result.window_ivs.push_back(result.inner_loop.getInductionVar()); in InsertWindowLoops()
612 window_loops.window_ivs, &inner_loop_b); in SelectIvs()