| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 51 assert(readOp.getShapedType().isa<TensorType>() && in bufferize()
98 assert(writeOp.getShapedType().isa<TensorType>() && in bufferize()
|
| H A D | VectorTransferSplitRewritePatterns.cpp | 192 int64_t memrefRank = xferOp.getShapedType().getRank(); in createSubViewIntersection()
257 if (compatibleMemRefType != xferOp.getShapedType()) in createFullPartialLinalgCopy()
312 if (compatibleMemRefType != xferOp.getShapedType()) in createFullPartialVectorTransferRead()
363 if (compatibleMemRefType != xferOp.getShapedType()) in getLocationToWriteFullVec()
572 getCastCompatibleMemRefType(xferOp.getShapedType().cast<MemRefType>(), in splitFullAndPartialTransfer()
|
| H A D | VectorTransferOpTransforms.cpp | 534 if (read.getShapedType().isa<MemRefType>()) in transferOpflowOpt()
539 if (write.getShapedType().isa<MemRefType>()) in transferOpflowOpt()
|
| H A D | VectorTransferPermutationMapRewritePatterns.cpp | 239 if (op.getShapedType().isa<TensorType>()) { in matchAndRewrite()
|
| H A D | VectorTransforms.cpp | 2056 auto memRefType = read.getShapedType().dyn_cast<MemRefType>(); in matchAndRewrite()
2195 auto memRefType = write.getShapedType().dyn_cast<MemRefType>(); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/VectorToSCF/ |
| H A D | VectorToSCF.cpp | 253 if (xferOp.getShapedType().template isa<RankedTensorType>()) { in isTensorOp()
494 Type type = isTensorOp(xferOp) ? xferOp.getShapedType() : Type(); in rewriteOp()
540 xferOp.getShapedType().getElementType()) in checkPrepareXferOp()
898 xferOp.getShapedType().getElementType()) in matchAndRewrite()
1025 xferOp.getShapedType().getElementType()) in matchAndRewrite()
1032 auto sourceType = isTensorOp(xferOp) ? xferOp.getShapedType() : Type(); in matchAndRewrite()
1228 auto memRefType = xferOp.getShapedType().template dyn_cast<MemRefType>(); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Interfaces/ |
| H A D | VectorInterfaces.td | 155 /*methodName=*/"getShapedType",
202 "return $_op.getShapedType().getRank() - $_op.getTransferRank();"
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | BuiltinAttributes.cpp | 882 static ShapedType getShapedType(MLIRContext *context, in getShapedType() function
890 static ShapedType getShapedType(MLIRContext *context, in getShapedType() function
898 static ShapedType getShapedType(MLIRContext *context, in getShapedType() function
906 static ShapedType getShapedType(MLIRContext *context, in getShapedType() function
914 static ShapedType getShapedType(MLIRContext *context, in getShapedType() function
922 static ShapedType getShapedType(MLIRContext *context, in getShapedType() function
934 auto shapedType = denseArrayAttrEltTypeBuilder<T>::getShapedType( in get()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/ |
| H A D | VectorToGPU.cpp | 117 if (!getMemrefConstantHorizontalStride(readOp.getShapedType())) in transferReadSupportsMMAMatrixType()
145 if (!getMemrefConstantHorizontalStride(writeOp.getShapedType())) in transferWriteSupportsMMAMatrixType()
398 getMemrefConstantHorizontalStride(op.getShapedType()); in convertTransferReadOp()
421 getMemrefConstantHorizontalStride(op.getShapedType()); in convertTransferWriteOp()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 2961 p << " : " << getShapedType() << ", " << getVectorType(); in print()
3027 ShapedType shapedType = getShapedType(); in verify()
3097 if (op.getShapedType().isDynamicDim(indicesIdx)) in isInBounds()
3156 if (!readOp.getShapedType().isa<RankedTensorType>()) in foldRAW()
3191 if (getShapedType().isa<MemRefType>()) in getEffects()
3317 !readOp.getShapedType().isa<RankedTensorType>()) in matchAndRewrite()
3481 p << " : " << getVectorType() << ", " << getShapedType(); in print()
3486 ShapedType shapedType = getShapedType(); in verify()
3629 if (getShapedType().isa<MemRefType>()) in getEffects()
3664 if (!writeOp.getShapedType().isa<RankedTensorType>()) in matchAndRewrite()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Hoisting.cpp | 411 if (!transferRead.getShapedType().isa<MemRefType>()) in hoistRedundantVectorTransfers()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 1822 rewriter, loc, typeConverter->convertType(transposeOp.getShapedType())); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 2701 if (getPermutation().getNumDims() != getShapedType().getRank()) in verify()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/MemRef/IR/ |
| H A D | MemRefOps.td | 1744 ShapedType getShapedType() { return getIn().getType().cast<ShapedType>(); }
|