Lines Matching refs:moduleOp
279 getFuncOpsOrderedByCalls(ModuleOp moduleOp, in getFuncOpsOrderedByCalls() argument
287 WalkResult res = moduleOp.walk([&](func::FuncOp funcOp) -> WalkResult { in getFuncOpsOrderedByCalls()
318 return moduleOp.emitOpError( in getFuncOpsOrderedByCalls()
357 mlir::bufferization::analyzeModuleOp(ModuleOp moduleOp, in analyzeModuleOp() argument
372 if (failed(getFuncOpsOrderedByCalls(moduleOp, orderedFuncOps, callerMap))) in analyzeModuleOp()
404 ModuleOp moduleOp, const OneShotAnalysisState &analysisState) { in bufferizeModuleOp() argument
409 IRRewriter rewriter(moduleOp.getContext()); in bufferizeModuleOp()
417 if (failed(getFuncOpsOrderedByCalls(moduleOp, orderedFuncOps, callerMap))) in bufferizeModuleOp()
433 moduleOp.walk([&](func::FuncOp op) { in bufferizeModuleOp()
442 ModuleOp moduleOp, const OneShotBufferizationOptions &options) { in runOneShotModuleBufferize() argument
445 OneShotAnalysisState analysisState(moduleOp, options); in runOneShotModuleBufferize()
446 if (failed(insertTensorCopies(moduleOp, options))) in runOneShotModuleBufferize()
450 if (failed(bufferizeModuleOp(moduleOp, analysisState))) in runOneShotModuleBufferize()