Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DPromotion.cpp134 DeallocBufferCallbackFn deallocationFn; member
175 if (options.deallocationFn) { in LinalgOpInstancePromotionOptions()
176 deallocationFn = *options.deallocationFn; in LinalgOpInstancePromotionOptions()
178 deallocationFn = [&](OpBuilder &b, Value buffer) { in LinalgOpInstancePromotionOptions()
356 (void)options.deallocationFn(b, pi.second.fullLocalView); in promoteSubViews()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizableOpInterface.cpp668 if (deallocationFn) in createDealloc()
669 return (*deallocationFn)(b, loc, allocatedBuffer); in createDealloc()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Bufferization/IR/
H A DBufferizableOpInterface.h220 Optional<DeallocationFn> deallocationFn; member
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/
H A DTransforms.h260 Optional<DeallocBufferCallbackFn> deallocationFn = None; member
265 deallocationFn = deallocFn; in setAllocationDeallocationFns()