Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_parallel_loops.cc492 scf::ParallelOp loop_over_src = in matchAndRewrite() local
494 rewriter.setInsertionPointToStart(loop_over_src.getBody()); in matchAndRewrite()
497 auto selected_ivs = SelectIvs(s_and_s_op, loop_over_src, &rewriter); in matchAndRewrite()
501 loop_over_src.getInductionVars()); in matchAndRewrite()
536 scf::ParallelOp loop_over_src, in InsertWindowLoops() argument
544 auto rank = loop_over_src.getNumLoops(); in InsertWindowLoops()
561 if (b->getInsertionBlock() == loop_over_src.getBody()) { in InsertWindowLoops()
601 scf::ParallelOp loop_over_src, in SelectIvs() argument
605 WindowLoops window_loops = InsertWindowLoops(s_and_s_op, loop_over_src, b); in SelectIvs()
611 MapWindowIvsToInput(s_and_s_op, loop_over_src.getInductionVars(), in SelectIvs()