Home
last modified time | relevance | path

Searched defs:dynamicSizes (Results 1 – 11 of 11) sorted by relevance

/llvm-project/mlir/lib/Conversion/LLVMCommon/
H A DPattern.cpp115 Location loc, MemRefType memRefType, ValueRange dynamicSizes, in getMemRefDescriptorSizes() argument
187 getNumElements(Location loc,MemRefType memRefType,ValueRange dynamicSizes,ConversionPatternRewriter & rewriter) const getNumElements() argument
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DConvertToDestinationStyle.cpp202 SmallVector<Value> dynamicSizes = reifyOrComputeDynamicSizes(rewriter, value); global() local
437 SmallVector<Value> dynamicSizes; rewriteInDestinationPassingStyle() local
/llvm-project/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizationOps.cpp262 build(builder, result, type, dynamicSizes, /*copy=*/Value(), in build() argument
269 build(OpBuilder & builder,OperationState & result,RankedTensorType type,ValueRange dynamicSizes,Value copy) build() argument
276 build(OpBuilder & builder,OperationState & result,TensorType type,ValueRange dynamicSizes,Value copy,IntegerAttr memorySpace) build() argument
H A DBufferizableOpInterface.cpp162 SmallVector<Value> dynamicSizes; in allocateTensorForShapedValue() local
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp179 foldDynamicToStaticDimSizes(RankedTensorType type,ValueRange dynamicSizes,SmallVector<Value> & foldedDynamicSizes) foldDynamicToStaticDimSizes() argument
883 build(OpBuilder & builder,OperationState & result,ArrayRef<int64_t> staticShape,Type elementType,ValueRange dynamicSizes,Attribute encoding) build() argument
892 SmallVector<Value> dynamicSizes; build() local
2035 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; inferResultType() local
2079 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; inferCanonicalRankReducedResultType() local
2097 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; build() local
2502 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; build() local
3453 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; build() local
3557 build(OpBuilder & builder,OperationState & result,Value element,Type aggregateType,ValueRange dynamicSizes) build() argument
3562 build(OpBuilder & builder,OperationState & result,Value element,ArrayRef<int64_t> staticShape,ValueRange dynamicSizes) build() argument
3570 SmallVector<Value> dynamicSizes; build() local
[all...]
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DBufferizableOpInterfaceImpl.cpp555 ValueRange dynamicSizes, in lowerGenerateLikeOpBody() argument
776 SmallVector<Value> dynamicSizes; bufferize() local
/llvm-project/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp1825 build(b, result, resultType, source, dynamicOffsets, dynamicSizes, in build() local
259 SmallVector<Value, 4> dynamicSizes; matchAndRewrite() local
2724 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; inferResultType() local
2776 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; inferRankReducedResultType() local
2794 SmallVector<Value> dynamicOffsets, dynamicSizes, dynamicStrides; build() local
[all...]
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/
H A DCodegenUtils.cpp398 SmallVector<Value> dynamicSizes; in allocDenseTensor() local
/llvm-project/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp2353 llvm::SmallVector<Value> dynamicSizes; matchAndRewrite() local
2463 SmallVector<Value> dynamicSizes; matchAndRewrite() local
[all...]
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseGPUCodegen.cpp152 SmallVector<Value> dynamicSizes; in genAllocMemRef() local
/llvm-project/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.cpp3008 ArrayRef<Operation *> dynamicSizes = dynamicSizeProducers[dynamicIdx]; apply() local