Lines Matching defs:loadOp

266   LogicalResult matchAndRewrite(OpTy loadOp,
276   LogicalResult matchAndRewrite(OpTy loadOp,
286   LogicalResult matchAndRewrite(OpTy loadOp,
427     OpTy loadOp, PatternRewriter &rewriter) const {
429       getMemRefOperand(loadOp).template getDefiningOp<memref::SubViewOp>();
432     return rewriter.notifyMatchFailure(loadOp, "not a subview producer");
435       preconditionsFoldSubViewOp(rewriter, loadOp, subViewOp);
439   SmallVector<Value> indices(loadOp.getIndices().begin(),
440                              loadOp.getIndices().end());
444           dyn_cast<affine::AffineLoadOp>(loadOp.getOperation())) {
447         affineMap, indices, loadOp.getLoc(), rewriter);
452       rewriter, loadOp.getLoc(), subViewOp.getMixedOffsets(),
456   llvm::TypeSwitch<Operation *, void>(loadOp)
459             loadOp, subViewOp.getSource(), sourceIndices);
463             loadOp, subViewOp.getSource(), sourceIndices, op.getNontemporal());
498     OpTy loadOp, PatternRewriter &rewriter) const {
500       getMemRefOperand(loadOp).template getDefiningOp<memref::ExpandShapeOp>();
505   SmallVector<Value> indices(loadOp.getIndices().begin(),
506                              loadOp.getIndices().end());
510           dyn_cast<affine::AffineLoadOp>(loadOp.getOperation())) {
513         affineMap, indices, loadOp.getLoc(), rewriter);
518           loadOp.getLoc(), rewriter, expandShapeOp, indices, sourceIndices)))
520   llvm::TypeSwitch<Operation *, void>(loadOp)
523             loadOp, expandShapeOp.getViewSource(), sourceIndices);
527             loadOp, expandShapeOp.getViewSource(), sourceIndices,
546     OpTy loadOp, PatternRewriter &rewriter) const {
547   auto collapseShapeOp = getMemRefOperand(loadOp)
553   SmallVector<Value> indices(loadOp.getIndices().begin(),
554                              loadOp.getIndices().end());
558           dyn_cast<affine::AffineLoadOp>(loadOp.getOperation())) {
561         affineMap, indices, loadOp.getLoc(), rewriter);
566           loadOp.getLoc(), rewriter, collapseShapeOp, indices, sourceIndices)))
568   llvm::TypeSwitch<Operation *, void>(loadOp)
571             loadOp, collapseShapeOp.getViewSource(), sourceIndices);
575             loadOp, collapseShapeOp.getViewSource(), sourceIndices,