NgdZddlmZgdZddlZddlZddlmZmZm Z ddl m Z ddl Z ddlmcmZddlZ ddl mZmZmZmZmZmZmZmZmZdZejZee_d e j !ie_"ej#Z$e d ed ef Z%d6dZ&efd7dZ'efd8dZ(ee$fd9d Z)ee$fd:d#Z*de$fd;d'Z+ee$fdd?d5Z.ej/Z/dS)@zDGraph 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)AnyCallableTypeVar)Message) IR_VERSIONAttributeProto FunctionProto GraphProto ModelProto NodeProtoSparseTensorProto TensorProtoValueInfoProtoi5wFuncType.)boundprotor proto_type type[Message]returnNonecVt||std|jddS)Nz'The proto message needs to be of type '') isinstance TypeError__name__)r!r"s H/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/checker.py_ensure_proto_typer,AsA eZ ( (  Lj6I L L L     value_inforctxC.CheckerContextczt|ttj||SN)r,rrr SerializeToString)r.r/s r+r r Hs3z>222 j::<&2244c : ::r-attrrlexical_scope_ctxC.LexicalScopeContextc|t|ttj|||Sr2)r,rrrr3)r6r/r7s r+rrTs6 t^,,, T3355sz"check_function..qs0     !>#9   r-) r,rrCheckerContextonnxhelperfind_min_ir_version_for opset_import ir_version opset_importsrr3)r<r/r7s r+rrfs x/// {  <<  !$=    "*"7   X//1138IJJJJJr-graphrc|t|ttj|||Sr2)r,rrrr3)rNr/r7s r+rrxs5 uj))) =0022C9J K KKr-sparserc~t|ttj||dSr2)r,rrr r3)rPr/s r+r r s:v0111&2244c:::::r-Fmodel&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. zZThis protobuf of onnx model is too large (>2GB). Call check_model with model path instead.N)r(strosPathLikercheck_model_pathfspathbytesr3sys getsizeofr ValueErrorr)rRrTrVrWprotobuf_strings r+rrs4%#r{+,,   Ie    *        u-- LEE53J3J3L3L  = ) ),< < <l    *       r-)r!rr"r#r$r%)r.rr/r0r$r%)r4rr/r0r$r%)r6rr/r0r7r8r$r%)r:rr/r0r7r8r$r%)r<rr/r=r7r8r$r%)rNrr/r0r7r8r$r%)rPrr/r0r$r%)FFF) rRrSrTrUrVrUrWrUr$r%)0__doc__ __future__r__all__rZr_typingrrrgoogle.protobuf.messager onnx.defsrHonnx.onnx_cpp2py_export.checkeronnx_cpp2py_exportcheckerronnx.shape_inferencerrrrrrrrrrrGr rLdefsonnx_opset_versionrMLexicalScopeContextr rr,r r rrrrr rrrAr-r+rpsKJ""""""     ))))))))))+++++++++++++++                      #!"$$'!#TY%A%A%C%C D--// 7:Xc3h%7 8 8 8    9HCCCCC?N;;;;;,/DOOOOO,/DJJJJJ$(/DKKKKK(,/DLLLLL8G;;;;;+0 % 0 0 0 0 0 f#r-