o XiH @shdZddlmZddlZddlmZmZmZddlm Z er"ddl Z e e Z dZedZdd dZdS)z+Utilities for interfacing with onnx C APIs.) annotationsN) TYPE_CHECKINGCallableTypeVar)iri_RfuncCallable[[onnx.ModelProto], _R]modelir.Modelreturnc Csjt|jj}dd|D}t|jj}|D]B}|jdus J|jdur*|jj|_|jdur4|jj|_||jjvrA|jj ||jj t krYd|_|j dusQJ|jj |j qtj|}z-||}W|D]}||j |_|j|qh|jjd|}|jj|jj||S|D]}||j |_|j|q|jjd|}|jj|jj|w)aCall an ONNX C API function by temporarily removing initializers. This is necessary because the ONNX C API does not support large models with initializers that have large tensor values. The input model is left unchanged no matter the call succeeds or not. Args: func: Partially applied function that takes a model proto and returns anything. model: The IR model to pass to the API function. Returns: The resulting ModelProto that contains the result of the API call. cSsi|]}|j|jqS)name const_value).0vr r ]/home/ubuntu/.local/lib/python3.10/site-packages/onnxscript/version_converter/_c_api_utils.py (sz!call_onnx_api..N)tuplegraph initializersvaluesleninputsrshapedtypeappendsize_BIG_TENSOR_SIZE_LIMITrpoprserdeserialize_modelregister_initializerclearextend) rr initializer_valuestensorsoriginal_inputs_len initializerprotoresultrr r r call_onnx_apis@             r+)rr r r r r)__doc__ __future__rloggingtypingrrr onnxscriptronnx getLogger__name__loggerrrr+r r r rs