/;jidZddlmZgdZddlZddlZddlmZddlZ ddl m cm Z ddlmZerddlmZdZe jZee_d e jie_e jZd3dZefd4dZefd5dZeefd6dZeefd7d Zdefd8d$Z eefd9d'Z!efd:d*Z" d;ds5vt/000 >&2244c : ::r!attronnx.AttributeProtolexical_scope_ctxC.LexicalScopeContextct|tjtj|||Sr')r r(AttributeProtorrr*)r/r$r1s rrrEs9 tT0111 T3355sz"check_function..bs0     !>#9   r!) r r( FunctionProtorCheckerContexthelperfind_min_ir_version_for opset_import ir_version opset_importsrr*)r9r$r1s rrrWs x!3444 {  <<  !$=    "*"7   X//1138IJJJJJr!graphonnx.GraphProtoct|tjtj|||Sr')r r( GraphProtorrr*)rLr$r1s rrris7 udo... =0022C9J K KKr!sparseonnx.SparseTensorProtoct|tjtj||dSr')r r(SparseTensorProtorr r*)rPr$s rr r rs<vt5666&2244c:::::r!Fmodel+onnx.ModelProto | str | bytes | os.PathLike full_checkboolskip_opset_compatibility_checkcheck_custom_domainczt|ttjfr+t jtj||||dSt|tr|n|}tj |tkrtdt j ||||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)rTrVrXrYprotobuf_strings rrrys4%#r{+,,   Ie    *        u-- LEE53J3J3L3L  = ) ),< < <m    *       r!)rrrrrr)r"r#r$r%rr)r+r,r$r%rr)r/r0r$r%r1r2rr)r5r6r$r%r1r2rr)r9r:r$r;r1r2rr)rLrMr$r%r1r2rr)rPrQr$r%rr)FFF) rTrUrVrWrXrWrYrWrr)%__doc__ __future__r__all__r\ratypingr onnx.defsr(onnx.onnx_cpp2py_export.checkeronnx_cpp2py_exportcheckerr onnx.onnx_pbrgoogle.protobuf.messagerrrFr rJdefsonnx_opset_versionrKLexicalScopeContextr r r r rrrrr rrr?r!rrrs9KJ""""""     +++++++++######0//////#!"$$'!#TY%A%A%C%C D--//    >MCCCCC7F;;;;;,/DOOOOO,/DJJJJJ$(/DKKKKK(,/DLLLLL=L;;;;;+0 % 0 0 0 0 0 f#r!