Lines Matching refs:execute
239 outlineExecuteOp(SymbolTable &symbolTable, ExecuteOp execute) { in outlineExecuteOp() argument
240 ModuleOp module = execute->getParentOfType<ModuleOp>(); in outlineExecuteOp()
243 Location loc = execute.getLoc(); in outlineExecuteOp()
247 cloneConstantsIntoTheRegion(execute.body()); in outlineExecuteOp()
250 SetVector<mlir::Value> functionInputs(execute.dependencies().begin(), in outlineExecuteOp()
251 execute.dependencies().end()); in outlineExecuteOp()
252 functionInputs.insert(execute.operands().begin(), execute.operands().end()); in outlineExecuteOp()
253 getUsedValuesDefinedAbove(execute.body(), functionInputs); in outlineExecuteOp()
259 auto outputTypes = execute.getResultTypes(); in outlineExecuteOp()
275 size_t numDependencies = execute.dependencies().size(); in outlineExecuteOp()
276 size_t numOperands = execute.operands().size(); in outlineExecuteOp()
293 valueMapping.map(execute.body().getArguments(), unwrappedOperands); in outlineExecuteOp()
297 for (Operation &op : execute.body().getOps()) in outlineExecuteOp()
328 ImplicitLocOpBuilder callBuilder(loc, execute); in outlineExecuteOp()
330 func.getName(), execute.getResultTypes(), functionInputs.getArrayRef()); in outlineExecuteOp()
331 execute.replaceAllUsesWith(callOutlinedFunc.getResults()); in outlineExecuteOp()
332 execute.erase(); in outlineExecuteOp()
743 module.walk([&](ExecuteOp execute) { in runOnOperation() argument
744 outlinedFunctions.insert(outlineExecuteOp(symbolTable, execute)); in runOnOperation()