Lines Matching defs:parallelOp

215   LogicalResult matchAndRewrite(mlir::scf::ParallelOp parallelOp,
482 ParallelLowering::matchAndRewrite(ParallelOp parallelOp,
484   Location loc = parallelOp.getLoc();
485   auto reductionOp = dyn_cast<ReduceOp>(parallelOp.getBody()->getTerminator());
495   SmallVector<Value, 4> iterArgs = llvm::to_vector<4>(parallelOp.getInitVals());
497   ivs.reserve(parallelOp.getNumLoops());
501        llvm::zip(parallelOp.getInductionVars(), parallelOp.getLowerBound(),
502                  parallelOp.getUpperBound(), parallelOp.getStep())) {
525   yieldOperands.reserve(parallelOp.getNumResults());
526   for (int64_t i = 0, e = parallelOp.getNumResults(); i < e; ++i) {
540     rewriter.mergeBlocks(parallelOp.getBody(), newBody, ivs);
542     rewriter.inlineBlockBefore(parallelOp.getBody(), newBody->getTerminator(),
552   rewriter.replaceOp(parallelOp, loopResults);