o öXœiSã@stdZddlmZddlZddlZddlmmZ dej deej ddfdd„Z d ej d eejdej fd d „ZdS) zPA utility function to replace custom operations in a model with their expansionsé)ÚSequenceNÚirmodelÚ irfunctionsÚreturncCsL|j}t|ƒdkr tdƒ‚|D]}||| ¡<qt ¡|ƒt ¡|ƒdS)a6A utility function to replace custom operations in a model with their expansions: The model is updated in-place. Args: irmodel: An ONNX model possibly containing calls to custom operations. irfunctions: A sequence of functions defining the expansions for the custom operations. rz.Input model cannot have model-local functions.N)Ú functionsÚlenÚ ValueErrorÚ identifierÚ common_passesÚ InlinePassÚRemoveUnusedOpsetsPass)rrÚmodel_functionsÚfunc©rúL/home/ubuntu/.local/lib/python3.10/site-packages/onnxscript/utils/replace.pyÚreplace_functions_inplace s  rÚmodelrcCs,t |¡}dd„|Dƒ}t||ƒt |¡S)auA utility function to replace custom operations in a model with their expansions: Args: model: An ONNX ModelProto possibly containing calls to custom operations. functions: A sequence of FunctionProto defining the expansions for the custom operations. Returns: An updated ModelProto with custom operations replaced by their expansions. cSsg|]}t |¡‘qSr)ÚirÚ from_proto)Ú.0rrrrÚ /sz%replace_functions..)rrrÚto_proto)rrrrrrrÚreplace_functions#s   r)Ú__doc__ÚtypingrÚonnxÚonnx_irrÚonnx_ir.passes.commonÚpassesÚcommonr ÚModelÚFunctionrÚ ModelProtoÚ FunctionProtorrrrrÚs ÿÿþ