| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/ |
| H A D | BufferizationOps.cpp | 137 if (shapedType.isDynamicDim(i)) { in populateDynamicDimSizes()
308 if (!op.isDynamicDim(i)) in matchAndRewrite()
338 if (!allocTensorOp.getType().isDynamicDim(*maybeConstantIndex)) in matchAndRewrite()
356 if (isDynamicDim(dim)) in reifyResultShapes()
412 assert(isDynamicDim(idx) && "expected dynamic dim"); in getDynamicSize()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalgNamed.cpp | 113 if (inputTy.isDynamicDim(i) && i != heightDim && i != weightDim) in inferDynamicDimsForConv()
118 if (inputTy.isDynamicDim(heightDim)) { in inferDynamicDimsForConv()
131 if (inputTy.isDynamicDim(weightDim)) { in inferDynamicDimsForConv()
514 if (!firstOperandTy.hasRank() || firstOperandTy.isDynamicDim(0)) { in matchAndRewrite()
518 if (!firstOperandTy.hasRank() || firstOperandTy.isDynamicDim(1)) { in matchAndRewrite()
522 if (!secondOperandTy.hasRank() || secondOperandTy.isDynamicDim(2)) { in matchAndRewrite()
579 if (!inputTy.hasRank() || inputTy.isDynamicDim(0)) { in matchAndRewrite()
583 if (!weightTy.hasRank() || weightTy.isDynamicDim(0)) { in matchAndRewrite()
|
| H A D | TosaToLinalg.cpp | 599 if (operandTy.isDynamicDim(i) && !dynDims[i]) in elementwiseMatchAndRewriteHelper()
802 if (inputTy.isDynamicDim(i)) in reduceMatchAndRewriteHelper()
1122 if (!operandTy.hasRank() || operandTy.isDynamicDim(index)) { in matchAndRewrite()
1664 if (inputType.isDynamicDim(i)) { in matchAndRewrite()
1723 if (inputTy.isDynamicDim(i)) { in matchAndRewrite()
1797 if (inputTy.isDynamicDim(i) || multiples[i] == -1) { in matchAndRewrite()
1944 if (inputTy.isDynamicDim(i) && i != axis) { in matchAndRewrite()
2116 if (inputTy.isDynamicDim(i)) { in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/BufferizationToMemRef/ |
| H A D | BufferizationToMemRef.cpp | 57 if (!memrefType.isDynamicDim(i)) in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Bufferization/IR/ |
| H A D | BufferizationOps.td | 105 bool isDynamicDim(unsigned idx) {
106 return getType().isDynamicDim(idx);
114 assert(isDynamicDim(idx) && "expected dynamic size");
128 assert(!isDynamicDim(idx) && "expected static size");
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | ExpandOps.cpp | 99 if (op.getType().isDynamicDim(i)) { in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Vectorization.cpp | 712 if (!sourceType.isDynamicDim(i)) { in tryVectorizeCopy()
718 } else if (!resultType.isDynamicDim(i)) { in tryVectorizeCopy()
939 if (t1.isDynamicDim(i) != t2.isDynamicDim(i)) in hasSameTensorSize()
941 if (!t1.isDynamicDim(i) && t1.getDimSize(i) != t2.getDimSize(i)) in hasSameTensorSize()
965 if (!t1.isDynamicDim(i)) in hasSameTensorSize()
|
| H A D | Transforms.cpp | 769 if (resultType.isDynamicDim(dim)) { in matchAndRewrite()
797 if (sourceType.isDynamicDim(dim)) { in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 176 if (one.isDynamicDim(i)) { in joinShapes()
180 if (two.isDynamicDim(i)) { in joinShapes()
336 if (!tensorType.isDynamicDim(index.getInt())) { in fold()
562 if (getType().isDynamicDim(dim)) { in reifyResultShapes()
1441 if (rankedTensorType.isDynamicDim(i)) in createCanonicalRankReducingExtractSliceOp()
1794 if (rankedTensorType.isDynamicDim(i)) in createCanonicalRankReducingInsertSliceOp()
1832 if (expectedType.isDynamicDim(i)) in verify()
1877 if (sourceType.isDynamicDim(i) || in inferResultType()
|
| /llvm-project-15.0.7/mlir/include/mlir/Interfaces/ |
| H A D | InferTypeOpInterface.h | 66 bool isDynamicDim(int index) const { in isDynamicDim() function
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/ |
| H A D | TosaOps.cpp | 387 if (i == axis || operandShape.isDynamicDim(i)) in inferReturnTypeComponents()
410 if (!operandShape.hasRank() || operandShape.isDynamicDim(axis)) { in inferReturnTypeComponents()
501 if (paddingShape.isDynamicDim(0)) { in inferReturnTypeComponents()
526 if (inputShape.isDynamicDim(i)) { in inferReturnTypeComponents()
653 permsShape.isDynamicDim(0)) { in inferReturnTypeComponents()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | Utils.cpp | 1555 bool isDynamicDim = false; in isNormalizedMemRefDynamicDim() local
1559 expr.walk([&inMemrefTypeDynDims, &isDynamicDim, &context](AffineExpr e) { in isNormalizedMemRefDynamicDim()
1563 isDynamicDim = true; in isNormalizedMemRefDynamicDim()
1568 return isDynamicDim; in isNormalizedMemRefDynamicDim()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/AMX/Transforms/ |
| H A D | LegalizeForLLVMExport.cpp | 63 if (mType.isDynamicDim(last)) { in getStride()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Shape/IR/ |
| H A D | Shape.cpp | 786 castOp.getType().cast<RankedTensorType>().isDynamicDim(0); in matchAndRewrite()
813 if (!resultTy || !resultTy.isDynamicDim(0)) in matchAndRewrite()
822 if (extentTensorTy.isDynamicDim(0)) in matchAndRewrite()
1606 if (!argTy || (!ty.isDynamicDim(0) && ty.getDimSize(0) != argTy.getRank())) in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/CAPI/IR/ |
| H A D | BuiltinTypes.cpp | 144 return unwrap(type).cast<ShapedType>().isDynamicDim( in mlirShapedTypeIsDynamicDim()
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | BuiltinTypes.td | 553 using ShapedType::Trait<MemRefType>::isDynamicDim;
724 using ShapedType::Trait<RankedTensorType>::isDynamicDim;
863 using ShapedType::Trait<UnrankedMemRefType>::isDynamicDim;
916 using ShapedType::Trait<UnrankedTensorType>::isDynamicDim;
|
| H A D | BuiltinTypeInterfaces.td | 149 bool isDynamicDim(unsigned idx) const {
|
| /llvm-project-15.0.7/mlir/lib/Conversion/ShapeToStandard/ |
| H A D | ShapeToStandard.cpp | 560 if (rankedTensorTy.isDynamicDim(i)) { in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 1791 if (sourceType.isDynamicDim(i)) in populateMap()
1829 !sourceType.isDynamicDim(i)) { in createNewOperandWithStaticSizes()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 476 memrefType.isDynamicDim(i) in bufferize()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | Utils.cpp | 589 if (memRefType.isDynamicDim(r)) in compute()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/ |
| H A D | GPUDialect.cpp | 1380 if (!memrefType || !memrefType.isDynamicDim(index.value())) in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseTensorConversion.cpp | 638 if (resType.isDynamicDim(i)) { in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 430 if (memRefType.isDynamicDim(i)) { in extractSizeOfRankedMemRef()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 885 if (!memrefType.isDynamicDim(index.getInt())) { in fold()
|