Lines Matching refs:adaptor
219 matchAndRewrite(memref::AllocaScopeOp allocaScopeOp, OpAdaptor adaptor, in matchAndRewrite()
276 matchAndRewrite(memref::AssumeAlignmentOp op, OpAdaptor adaptor, in matchAndRewrite()
278 Value memref = adaptor.getMemref(); in matchAndRewrite()
328 matchAndRewrite(memref::DeallocOp op, OpAdaptor adaptor, in matchAndRewrite()
332 MemRefDescriptor memref(adaptor.getMemref()); in matchAndRewrite()
348 matchAndRewrite(memref::DimOp dimOp, OpAdaptor adaptor, in matchAndRewrite()
354 operandType, dimOp, adaptor.getOperands(), rewriter)}); in matchAndRewrite()
361 adaptor.getOperands(), rewriter)}); in matchAndRewrite()
369 OpAdaptor adaptor, in extractSizeOfUnrankedMemRef()
381 UnrankedMemRefDescriptor unrankedDesc(adaptor.getSource()); in extractSizeOfUnrankedMemRef()
402 loc, createIndexConstant(rewriter, loc, 1), adaptor.getIndex()); in extractSizeOfUnrankedMemRef()
422 OpAdaptor adaptor, in extractSizeOfRankedMemRef()
432 MemRefDescriptor descriptor(adaptor.getSource()); in extractSizeOfRankedMemRef()
439 Value index = adaptor.getIndex(); in extractSizeOfRankedMemRef()
441 MemRefDescriptor memrefDescriptor(adaptor.getSource()); in extractSizeOfRankedMemRef()
493 matchAndRewrite(memref::GenericAtomicRMWOp atomicOp, OpAdaptor adaptor, in matchAndRewrite()
513 auto dataPtr = getStridedElementPtr(loc, memRefType, adaptor.getMemref(), in matchAndRewrite()
514 adaptor.getIndices(), rewriter); in matchAndRewrite()
601 matchAndRewrite(memref::GlobalOp global, OpAdaptor adaptor, in matchAndRewrite()
694 matchAndRewrite(memref::LoadOp loadOp, OpAdaptor adaptor, in matchAndRewrite()
699 getStridedElementPtr(loadOp.getLoc(), type, adaptor.getMemref(), in matchAndRewrite()
700 adaptor.getIndices(), rewriter); in matchAndRewrite()
712 matchAndRewrite(memref::StoreOp op, OpAdaptor adaptor, in matchAndRewrite()
716 Value dataPtr = getStridedElementPtr(op.getLoc(), type, adaptor.getMemref(), in matchAndRewrite()
717 adaptor.getIndices(), rewriter); in matchAndRewrite()
718 rewriter.replaceOpWithNewOp<LLVM::StoreOp>(op, adaptor.getValue(), dataPtr); in matchAndRewrite()
729 matchAndRewrite(memref::PrefetchOp prefetchOp, OpAdaptor adaptor, in matchAndRewrite()
734 Value dataPtr = getStridedElementPtr(loc, type, adaptor.getMemref(), in matchAndRewrite()
735 adaptor.getIndices(), rewriter); in matchAndRewrite()
758 matchAndRewrite(memref::RankOp op, OpAdaptor adaptor, in matchAndRewrite()
763 UnrankedMemRefDescriptor desc(adaptor.getMemref()); in matchAndRewrite()
803 void rewrite(memref::CastOp memRefCastOp, OpAdaptor adaptor, in rewrite()
812 return rewriter.replaceOp(memRefCastOp, {adaptor.getSource()}); in rewrite()
823 loc, adaptor.getSource(), rewriter); in rewrite()
844 UnrankedMemRefDescriptor memRefDesc(adaptor.getSource()); in rewrite()
871 lowerToMemCopyIntrinsic(memref::CopyOp op, OpAdaptor adaptor, in lowerToMemCopyIntrinsic()
876 MemRefDescriptor srcDesc(adaptor.getSource()); in lowerToMemCopyIntrinsic()
896 MemRefDescriptor targetDesc(adaptor.getTarget()); in lowerToMemCopyIntrinsic()
912 lowerToMemCopyFunctionCall(memref::CopyOp op, OpAdaptor adaptor, in lowerToMemCopyFunctionCall()
936 ? makeUnranked(adaptor.getSource(), srcType) in lowerToMemCopyFunctionCall()
937 : adaptor.getSource(); in lowerToMemCopyFunctionCall()
939 ? makeUnranked(adaptor.getTarget(), targetType) in lowerToMemCopyFunctionCall()
940 : adaptor.getTarget(); in lowerToMemCopyFunctionCall()
970 matchAndRewrite(memref::CopyOp op, OpAdaptor adaptor, in matchAndRewrite()
987 return lowerToMemCopyIntrinsic(op, adaptor, rewriter); in matchAndRewrite()
989 return lowerToMemCopyFunctionCall(op, adaptor, rewriter); in matchAndRewrite()
1041 matchAndRewrite(memref::ReinterpretCastOp castOp, OpAdaptor adaptor, in matchAndRewrite()
1047 adaptor, &descriptor))) in matchAndRewrite()
1057 memref::ReinterpretCastOp::Adaptor adaptor, Value *descriptor) const { in convertSourceMemRefToDescriptor()
1072 castOp.getSource(), adaptor.getSource(), in convertSourceMemRefToDescriptor()
1079 desc.setOffset(rewriter, loc, adaptor.getOffsets()[0]); in convertSourceMemRefToDescriptor()
1088 desc.setSize(rewriter, loc, i, adaptor.getSizes()[dynSizeId++]); in convertSourceMemRefToDescriptor()
1093 desc.setStride(rewriter, loc, i, adaptor.getStrides()[dynStrideId++]); in convertSourceMemRefToDescriptor()
1107 matchAndRewrite(memref::ReshapeOp reshapeOp, OpAdaptor adaptor, in matchAndRewrite()
1113 adaptor, &descriptor))) in matchAndRewrite()
1123 memref::ReshapeOp::Adaptor adaptor, in convertSourceMemRefToDescriptor()
1143 reshapeOp.getSource(), adaptor.getSource(), in convertSourceMemRefToDescriptor()
1208 MemRefDescriptor shapeDesc(adaptor.getShape()); in convertSourceMemRefToDescriptor()
1232 reshapeOp.getSource(), adaptor.getSource(), in convertSourceMemRefToDescriptor()
1578 matchAndRewrite(ReshapeOp reshapeOp, typename ReshapeOp::Adaptor adaptor, in matchAndRewrite() argument
1590 MemRefDescriptor srcDesc(adaptor.getSrc()); in matchAndRewrite()
1642 matchAndRewrite(memref::SubViewOp subViewOp, OpAdaptor adaptor, in matchAndRewrite()
1675 if (!LLVM::isCompatibleType(adaptor.getOperands().front().getType())) in matchAndRewrite()
1677 MemRefDescriptor sourceMemRef(adaptor.getOperands().front()); in matchAndRewrite()
1722 ? adaptor.getOperands()[subViewOp.getIndexOfDynamicOffset(i)] in matchAndRewrite()
1772 ? adaptor.getOperands()[subViewOp.getIndexOfDynamicSize(i)] in matchAndRewrite()
1782 ? adaptor.getOperands()[subViewOp.getIndexOfDynamicStride(i)] in matchAndRewrite()
1812 matchAndRewrite(memref::TransposeOp transposeOp, OpAdaptor adaptor, in matchAndRewrite() argument
1815 MemRefDescriptor viewMemRef(adaptor.getIn()); in matchAndRewrite()
1891 matchAndRewrite(memref::ViewOp viewOp, OpAdaptor adaptor, in matchAndRewrite()
1919 MemRefDescriptor sourceMemRef(adaptor.getSource()); in matchAndRewrite()
1935 loc, alignedPtr.getType(), alignedPtr, adaptor.getByteShift()); in matchAndRewrite()
1958 adaptor.getSizes(), i); in matchAndRewrite()
2008 matchAndRewrite(memref::AtomicRMWOp atomicOp, OpAdaptor adaptor, in matchAndRewrite()
2015 auto resultType = adaptor.getValue().getType(); in matchAndRewrite()
2018 getStridedElementPtr(atomicOp.getLoc(), memRefType, adaptor.getMemref(), in matchAndRewrite()
2019 adaptor.getIndices(), rewriter); in matchAndRewrite()
2021 atomicOp, resultType, *maybeKind, dataPtr, adaptor.getValue(), in matchAndRewrite()