Lines Matching defs:rewriter

58 applyTileAndFuseToAll(RewriterBase &rewriter, Operation *transformOp,
103     rewriter.setInsertionPoint(target);
105         scf::tileConsumerAndFuseProducersUsingSCF(rewriter, tilingInterfaceOp,
117           rewriter.replaceUsesWithIf(res, replacement, [&](OpOperand &use) {
125         rewriter.eraseOp(toReplace);
146 transform::TestFuseAndYieldOp::apply(TransformRewriter &rewriter,
155       getAsIndexOpFoldResult(rewriter.getContext(), tileSizes);
158       rewriter, getOperation(), state.getPayloadOps(getTarget()),
173 applyFuseConsumer(RewriterBase &rewriter, Operation *transformOp,
180     rewriter.setInsertionPoint(target);
184           scf::tileAndFuseConsumerOfSlice(rewriter, target);
203 transform::TestFuseConsumerOp::apply(TransformRewriter &rewriter,
207       rewriter, getOperation(), state.getPayloadOps(getTarget()),
228 applyTileToAll(RewriterBase &rewriter, Operation *transformOp,
246     rewriter.setInsertionPoint(target);
248         scf::tileUsingSCF(rewriter, tilingInterfaceOp, tilingOptions);
253     rewriter.replaceOp(tilingInterfaceOp,
270 transform::TestTileUsingForallOp::apply(TransformRewriter &rewriter,
278       getAsIndexOpFoldResult(rewriter.getContext(), tileSizes);
281       applyTileToAll(rewriter, getOperation(), state.getPayloadOps(getTarget()),
302     RewriterBase &rewriter, Operation *transformOp, Range &&payloadOps,
314     rewriter.setInsertionPoint(target);
326           rewriter.replaceAllUsesWith(res, replacement);
328         rewriter.eraseOp(toReplace);
348 transform::TestFuseUsingForallOp::apply(TransformRewriter &rewriter,
359       getAsIndexOpFoldResult(rewriter.getContext(), tileSizes);
365       rewriter, getOperation(), state.getPayloadOps(getRootOp()),
369         return tileConsumerAndFuseProducersUsingSCF(rewriter, tilingInterfaceOp,