Searched refs:allocationFn (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Promotion.cpp | 133 AllocBufferCallbackFn allocationFn; member
164 if (options.allocationFn) { in LinalgOpInstancePromotionOptions()
165 allocationFn = *options.allocationFn; in LinalgOpInstancePromotionOptions()
167 allocationFn = [&](OpBuilder &b, memref::SubViewOp subViewOp, in LinalgOpInstancePromotionOptions()
213 const AllocBufferCallbackFn &allocationFn, DataLayout &layout) { in promoteSubviewAsNewBuffer() argument
242 Optional<Value> fullLocalView = allocationFn(b, subView, fullSizes, layout); in promoteSubviewAsNewBuffer()
264 b, b.getLoc(), subView, options.allocationFn, layout); in promoteSubViews()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| H A D | Transforms.h | 259 Optional<AllocBufferCallbackFn> allocationFn = None; member
264 allocationFn = allocFn; in setAllocationDeallocationFns()
291 const AllocBufferCallbackFn &allocationFn,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/ |
| H A D | BufferizableOpInterface.cpp | 652 if (allocationFn) in createAlloc()
653 return (*allocationFn)(b, loc, type, dynShape, bufferAlignment); in createAlloc()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Bufferization/IR/ |
| H A D | BufferizableOpInterface.h | 219 Optional<AllocationFn> allocationFn; member
|