| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 969 assert(static_cast<int64_t>(staticSizes.size()) == in inferResultType()
972 return RankedTensorType::get(staticSizes, in inferResultType()
979 SmallVector<int64_t> staticOffsets, staticSizes, staticStrides; in inferResultType() local
983 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in inferResultType()
1027 SmallVector<int64_t> staticOffsets, staticSizes, staticStrides; in inferCanonicalRankReducedResultType() local
1031 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in inferCanonicalRankReducedResultType()
1048 SmallVector<int64_t> staticOffsets, staticSizes, staticStrides; in build() local
1052 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in build()
1467 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in build()
1500 extractFromI64ArrayAttr(staticSizes), in verifyInsertSliceOp()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Interfaces/ |
| H A D | ViewLikeInterface.cpp | 198 mlir::getMixedSizes(OffsetSizeAndStrideOpInterface op, ArrayAttr staticSizes, in getMixedSizes() argument
202 unsigned count = static_cast<unsigned>(staticSizes.size()); in getMixedSizes()
207 res.push_back(staticSizes[idx]); in getMixedSizes()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 1193 SmallVector<int64_t, 4> staticSizes; in build() local
1194 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in build()
1196 auto resultType = RankedTensorType ::get(staticSizes, elementType); in build()
1216 staticSizes, resultType.getElementType(), resultType.getEncoding()); in verify()
1225 Type InitTensorOp::inferResultType(ArrayRef<int64_t> staticSizes, in inferResultType() argument
1227 return RankedTensorType::get(staticSizes, elementType, encoding); in inferResultType()
1263 SmallVector<int64_t, 4> staticSizes; in matchAndRewrite() local
1267 staticSizes.push_back(op.getStaticSize(i)); in matchAndRewrite()
1277 staticSizes.push_back(constantIndexOp.value()); in matchAndRewrite()
1283 staticSizes.push_back(ShapedType::kDynamicSize); in matchAndRewrite()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 1473 SmallVector<int64_t> staticOffsets, staticSizes, staticStrides; in build() local
1477 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in build()
2088 ArrayRef<int64_t> staticSizes, in inferResultType() argument
2129 staticSizes, sourceMemRefType.getElementType(), in inferResultType()
2139 SmallVector<int64_t> staticOffsets, staticSizes, staticStrides; in inferResultType() local
2143 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in inferResultType()
2148 staticSizes, staticStrides); in inferResultType()
2184 SmallVector<int64_t> staticOffsets, staticSizes, staticStrides; in inferRankReducedResultType() local
2188 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in inferRankReducedResultType()
2205 SmallVector<int64_t> staticOffsets, staticSizes, staticStrides; in build() local
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Interfaces/ |
| H A D | ViewLikeInterface.h | 43 ArrayAttr staticSizes,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 964 ArrayAttr staticSizes; in parse() local
968 parseOperandsOrIntegersSizesList(parser, dynamicSizes, staticSizes) || in parse()
973 result.addAttribute(getStaticSizesAttrName(result.name), staticSizes); in parse()
975 staticSizes.size() - llvm::count(extractFromI64ArrayAttr(staticSizes), 0); in parse()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Transforms.cpp | 767 SmallVector<int64_t> staticSizes; in matchAndRewrite() local
779 staticSizes.push_back(resultType.getDimSize(dim)); in matchAndRewrite()
784 padOp.getLoc(), dynSizes, staticSizes, resultType.getElementType()); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/ |
| H A D | LinalgOps.td | 62 static Type inferResultType(ArrayRef<int64_t> staticSizes, Type elementType,
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Tensor/IR/ |
| H A D | TensorOps.td | 311 ArrayRef<int64_t> staticSizes,
316 ArrayRef<OpFoldResult> staticSizes,
331 ArrayRef<int64_t> staticSizes,
337 ArrayRef<OpFoldResult> staticSizes,
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/MemRef/IR/ |
| H A D | MemRefOps.td | 1642 ArrayRef<int64_t> staticSizes,
1646 ArrayRef<OpFoldResult> staticSizes,
1659 ArrayRef<int64_t> staticSizes,
1664 ArrayRef<OpFoldResult> staticSizes,
|