U q±Ëhºã@sTddlmZddlZddlmmZd ddddœdd„Zddd ddd œd d „ZdS)é)Ú annotationsNFzonnx.ModelProtoÚbool)ÚmodelÚconvert_versionÚreturncCs&t | ¡|¡}t ¡}| |¡|S)abInline model-local functions in given model. Arguments: model: an ONNX ModelProto convert_version: if true, try to apply automatic version-conversion to functions requiring a different (ONNX) opset version from the model. Returns: ModelProto with all calls to model-local functions inlined (recursively) )ÚCÚinline_local_functionsÚSerializeToStringÚonnxÚ ModelProtoÚParseFromString)rrÚresultÚ inlined_model©rú0/tmp/pip-unpacked-wheel-xnis5xre/onnx/inliner.pyr s  rzlist[tuple[str, str]])rÚ function_idsÚexcludercCs(t | ¡||¡}t ¡}| |¡|S)aüInline selected model-local functions in given model. Arguments: model: an ONNX ModelProto function_ids: list of functions to include/exclude when inlining. Each element is a tuple of (function domain, function name). exclude: if true, inlines all functions except those specified in function_ids. if false, inlines all functions specified in function_ids. Returns: ModelProto with all calls to model-local functions inlined (recursively) )rÚinline_selected_functionsr r r r )rrrr rrrrrsÿ r)F)F) Ú __future__rr Zonnx.onnx_cpp2py_export.inlinerZonnx_cpp2py_exportZinlinerrrrrrrrÚs ÿÿ