| /llvm-project-15.0.7/mlir/lib/Conversion/LLVMCommon/ |
| H A D | Pattern.cpp | 121 Location loc, MemRefType memRefType, ValueRange dynamicSizes, in getMemRefDescriptorSizes() argument
127 static_cast<ssize_t>(dynamicSizes.size()) && in getMemRefDescriptorSizes()
134 ? dynamicSizes[dynamicIndex++] in getMemRefDescriptorSizes()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/ |
| H A D | BufferizationOps.cpp | 273 RankedTensorType type, ValueRange dynamicSizes) { in build() argument
274 build(builder, result, type, dynamicSizes, /*copy=*/Value(), in build()
279 RankedTensorType type, ValueRange dynamicSizes, in build() argument
281 build(builder, result, type, dynamicSizes, copy, in build()
|
| H A D | BufferizableOpInterface.cpp | 79 SmallVector<Value> dynamicSizes; in allocateTensorForShapedValue() local
95 dynamicSizes.push_back(shape[dim.index()]); in allocateTensorForShapedValue()
102 populateDynamicDimSizes(b, loc, tensor, dynamicSizes); in allocateTensorForShapedValue()
106 auto allocTensorOp = b.create<AllocTensorOp>(loc, tensorType, dynamicSizes, in allocateTensorForShapedValue()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 980 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; in inferResultType() local
983 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in inferResultType()
1028 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; in inferCanonicalRankReducedResultType() local
1031 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in inferCanonicalRankReducedResultType()
1049 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; in build() local
1052 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in build()
1464 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; in build() local
1467 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in build()
2256 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; in build() local
2259 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in build()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 172 SmallVector<Value, 4> dynamicSizes; in matchAndRewrite() local
191 dynamicSizes.push_back(dynamicSize); in matchAndRewrite()
199 assert(static_cast<int64_t>(dynamicSizes.size()) == in matchAndRewrite()
1474 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; in build() local
1477 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in build()
2140 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; in inferResultType() local
2143 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in inferResultType()
2185 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; in inferRankReducedResultType() local
2188 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in inferRankReducedResultType()
2206 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; in build() local
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Conversion/LLVMCommon/ |
| H A D | Pattern.h | 98 ValueRange dynamicSizes,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 1192 SmallVector<Value, 4> dynamicSizes; in build() local
1194 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in build()
1197 build(b, result, resultType, dynamicSizes, b.getI64ArrayAttr(staticSizes)); in build()
1262 SmallVector<Value, 4> dynamicSizes; in matchAndRewrite() local
1282 dynamicSizes.push_back(sizeOperand); in matchAndRewrite()
1290 rewriter.create<InitTensorOp>(op.getLoc(), newType, dynamicSizes, in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 963 SmallVector<OpAsmParser::UnresolvedOperand> dynamicSizes; in parse() local
968 parseOperandsOrIntegersSizesList(parser, dynamicSizes, staticSizes) || in parse()
969 parser.resolveOperands(dynamicSizes, pdlOperationType, result.operands) || in parse()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Bufferization/IR/ |
| H A D | BufferizationOps.td | 135 OpBuilder<(ins "RankedTensorType":$type, "ValueRange":$dynamicSizes)>,
138 OpBuilder<(ins "RankedTensorType":$type, "ValueRange":$dynamicSizes,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseTensorConversion.cpp | 333 SmallVector<Value> dynamicSizes; in allocDenseTensor() local
336 dynamicSizes.push_back(sizes[i]); in allocDenseTensor()
338 Value mem = builder.create<memref::AllocOp>(loc, memTp, dynamicSizes); in allocDenseTensor()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/MemRef/IR/ |
| H A D | MemRefOps.td | 65 let arguments = (ins Variadic<Index>:$dynamicSizes,
78 OpBuilder<(ins "MemRefType":$memrefType, "ValueRange":$dynamicSizes,
80 return build($_builder, $_state, memrefType, dynamicSizes, {}, alignment);
82 OpBuilder<(ins "MemRefType":$memrefType, "ValueRange":$dynamicSizes,
86 $_state.addOperands(dynamicSizes);
90 static_cast<int32_t>(dynamicSizes.size()),
103 `(`$dynamicSizes`)` (`` `[` $symbolOperands^ `]`)? attr-dict `:` type($memref)
|
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/ |
| H A D | GPUDialect.cpp | 1350 if (static_cast<int64_t>(dynamicSizes().size()) != in verify()
1387 Value substituteOp = *(alloc.dynamicSizes().begin() + in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/ |
| H A D | GPUToLLVMConversion.cpp | 462 getMemRefDescriptorSizes(loc, memRefType, adaptor.dynamicSizes(), rewriter, in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 1861 ArrayRef<int64_t> shape, ValueRange dynamicSizes, in getSize()
1869 return dynamicSizes[nDynamic]; in getSize()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/GPU/IR/ |
| H A D | GPUOps.td | 936 Variadic<Index>:$dynamicSizes, Variadic<Index>:$symbolOperands);
946 `(` $dynamicSizes `)` (`` `[` $symbolOperands^ `]`)? attr-dict `:` type($memref)
|