o Ďi@s$dZddlmZgdZddlZddlZddlmZddlZ ddl m m Z ddlmZer5ddlmZdZe Zee_d e jie_e ZdddZeefd?d d!Zeefd@d$d%ZdefdAd)d*Z eefdBd-d.Z!efdCd1d2Z" 3 3 3dDdEd:d;Z#e j$Z$dS)FzDGraph utilities for checking whether an ONNX proto message is legal.) annotations) check_attributecheck_function check_graph check_model check_nodecheck_sparse_tensor check_tensorcheck_value_infoDEFAULT_CONTEXTLEXICAL_SCOPE_CONTEXTValidationErrorCMAXIMUM_PROTOBUFN) TYPE_CHECKING) IR_VERSION)Messagelprotor proto_type type[Message]returnNonecCs t||std|jddS)Nz'The proto message needs to be of type '') isinstance TypeError__name__)rrrH/home/jeff/fluffinator/venv/lib/python3.10/site-packages/onnx/checker.py_ensure_proto_type0s  r value_infoonnx.ValueInfoProtoctxC.CheckerContextcCt|tjt||SN)ronnxValueInfoProtorr SerializeToString)r r"rrrr 7 r tensoronnx.TensorProtocCr$r%)rr& TensorProtorr r()r*r"rrrr >r)r attronnx.AttributeProtolexical_scope_ctxC.LexicalScopeContextcCt|tjt|||Sr%)rr&AttributeProtorrr()r-r"r/rrrrE rnodeonnx.NodeProtocCr1r%)rr& NodeProtorrr()r4r"r/rrrrNr3rfunctiononnx.FunctionProtoC.CheckerContext | NonecCsXt|tj|dur!t}tjj|jdd|_dd|jD|_ t | ||dS)NT)ignore_unknowncSsi|]}|j|jqSr)domainversion).0domain_versionrrr bsz"check_function..) rr& FunctionProtorCheckerContexthelperfind_min_ir_version_for opset_import ir_version opset_importsrr()r7r"r/rrrrWs rgraphonnx.GraphProtocCr1r%)rr& GraphProtorrr()rGr"r/rrrrir3rsparseonnx.SparseTensorProtocCs t|tjt||dSr%)rr&SparseTensorProtorrr()rJr"rrrrrs rFmodel+onnx.ModelProto | str | bytes | os.PathLike full_checkboolskip_opset_compatibility_checkcheck_custom_domaincCsjt|ttjfrtt||||dSt|tr|n|}t |t kr+t dt ||||dS)arCheck the consistency of a model. An exception will be raised if the model's ir_version is not set properly or is higher than checker's ir_version, or if the model has duplicate keys in metadata_props. If IR version >= 3, the model must specify opset_import. If IR version < 3, the model cannot have any opset_import specified. Args: model: Model to check. If model is a path, the function checks model path first. If the model bytes size is larger than 2GB, function should be called using model path. full_check: If True, the function also runs shape inference check. skip_opset_compatibility_check: If True, the function skips the check for opset compatibility. check_custom_domain: If True, the function will check all domains. Otherwise only check built-in domains. z[This protobuf of onnx model is too large (>2GiB). Call check_model with model path instead.N)rstrosPathLikercheck_model_pathfspathbytesr(sys getsizeofr ValueErrorr)rMrOrQrRprotobuf_stringrrrrys&r)rrrrrr)r r!r"r#rr)r*r+r"r#rr)r-r.r"r#r/r0rr)r4r5r"r#r/r0rr)r7r8r"r9r/r0rr)rGrHr"r#r/r0rr)rJrKr"r#rr)FFF) rMrNrOrPrQrPrRrPrr)%__doc__ __future__r__all__rTrYtypingr onnx.defsr&onnx.onnx_cpp2py_export.checkeronnx_cpp2py_exportcheckerr onnx.onnx_pbrgoogle.protobuf.messagerrrAr rEdefsonnx_opset_versionrFLexicalScopeContextr rr r rrrrrrr rrrrsP                 3