Home
last modified time | relevance | path

Searched refs:shapedValue (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizableOpInterface.cpp68 OpBuilder &b, Location loc, Value shapedValue, bool escape, in allocateTensorForShapedValue() argument
71 if (shapedValue.getType().isa<RankedTensorType>()) { in allocateTensorForShapedValue()
72 tensor = shapedValue; in allocateTensorForShapedValue()
73 } else if (shapedValue.getType().isa<MemRefType>()) { in allocateTensorForShapedValue()
74 tensor = b.create<ToTensorOp>(loc, shapedValue); in allocateTensorForShapedValue()
84 if (shapedValue.getType().isa<RankedTensorType>() && in allocateTensorForShapedValue()
85 shapedValue.isa<OpResult>()) { in allocateTensorForShapedValue()
87 shapedValue.getDefiningOp())) { in allocateTensorForShapedValue()
92 resultDims[shapedValue.cast<OpResult>().getResultNumber()]; in allocateTensorForShapedValue()
H A DBufferizationOps.cpp133 OpBuilder &b, Location loc, Value shapedValue, in populateDynamicDimSizes() argument
135 auto shapedType = shapedValue.getType().cast<ShapedType>(); in populateDynamicDimSizes()
139 dynamicDims.push_back(b.create<memref::DimOp>(loc, shapedValue, i)); in populateDynamicDimSizes()
142 dynamicDims.push_back(b.create<tensor::DimOp>(loc, shapedValue, i)); in populateDynamicDimSizes()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Bufferization/IR/
H A DBufferization.h38 void populateDynamicDimSizes(OpBuilder &b, Location loc, Value shapedValue,
H A DBufferizableOpInterface.h478 allocateTensorForShapedValue(OpBuilder &b, Location loc, Value shapedValue,