Searched refs:boundingSubViewSize (Results 1 – 3 of 3) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Promotion.cpp | 108 SubViewOp subView, ArrayRef<Value> boundingSubViewSize, bool dynamicBuffers, in defaultAllocBufferCallBack() argument
113 assert(rank > 0 && boundingSubViewSize.size() == static_cast<size_t>(rank)); in defaultAllocBufferCallBack()
118 for (auto size : llvm::enumerate(boundingSubViewSize)) in defaultAllocBufferCallBack()
122 SmallVector<int64_t, 4> dynSizes(boundingSubViewSize.size(), in defaultAllocBufferCallBack()
126 zero, boundingSubViewSize); in defaultAllocBufferCallBack()
193 ArrayRef<Value> boundingSubViewSize, in LinalgOpInstancePromotionOptions()
196 boundingSubViewSize, dynamicBuffers, in LinalgOpInstancePromotionOptions()
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestLinalgTransforms.cpp | 249 ArrayRef<Value> boundingSubViewSize, in allocCallBackFn() argument
251 SmallVector<int64_t, 4> shape(boundingSubViewSize.size(), -1); in allocCallBackFn()
257 boundingSubViewSize) in allocCallBackFn()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| D | Transforms.h | 150 OpBuilder &b, SubViewOp subView, ArrayRef<Value> boundingSubViewSize,
|