Lines Matching defs:forallOp

493   scf::ForallOp forallOp;
504     forallOp = rewriter.create<scf::ForallOp>(loc, nonZeroNumThreads,
510     forallOp = rewriter.create<scf::ForallOp>(loc, lbs, ubs, steps,
513   loops.push_back(forallOp);
515   rewriter.setInsertionPoint(forallOp.getTerminator());
516   destinationTensors = forallOp.getRegionOutArgs();
520   if (failed(tiledBodyFn(rewriter, loc, forallOp.getInductionVars(),
525   rewriter.setInsertionPointToEnd(forallOp.getTerminator().getBody());
1928   auto forallOp = dyn_cast<scf::ForallOp>(containingOp);
1929   if (!forallOp)
1932       forallOp.getTiedOpResult(forallOp.getTiedOpOperand(iterArg))