Home
last modified time | relevance | path

Searched refs:allocationFn (Results 1 – 2 of 2) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DPromotion.cpp159 AllocBufferCallbackFn allocationFn; member
190 allocationFn = in LinalgOpInstancePromotionOptions()
191 (options.allocationFn ? *(options.allocationFn) in LinalgOpInstancePromotionOptions()
233 AllocBufferCallbackFn allocationFn, OperationFolder *folder) { in promoteSubviewAsNewBuffer() argument
252 Optional<Value> fullLocalView = allocationFn(b, subView, fullSizes, folder); in promoteSubviewAsNewBuffer()
277 b, loc, subView, options.allocationFn, folder); in promoteSubViews()
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/
DTransforms.h216 Optional<AllocBufferCallbackFn> allocationFn = None; member
221 allocationFn = allocFn; in setAllocationDeallocationFns()
248 AllocBufferCallbackFn allocationFn,