bIg:dZddlmZgdZddlZddlZddlmZmZm Z ddl m Z ddl Z ddlmcmZddlZ ddl mZmZmZmZmZmZmZmZmZdZej:Zee_d e j@jCie_"ejFZ$e d ed ef Z%dd Z&ef ddZ'efddZ(ee$f ddZ)ee$f ddZ*de$f ddZ+ee$f ddZ,ef ddZ- d ddZ.ej^Z/y) 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.)boundcNt||std|jdy)Nz'The proto message needs to be of type '') isinstance TypeError__name__)proto proto_types I/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/checker.py_ensure_proto_typer)As1 eZ (5j6I6I5J! L   )cjt|ttj|j |SN)r)rrr SerializeToString) value_infoctxs r(r r Hs*z>2  j::>&224c ::r*clt|ttj|j ||Sr,)r)rrrr-)attrr/lexical_scope_ctxs r(rrTs- t^,  T335s#9#9 9 X//138IJ  s%B1clt|ttj|j ||Sr,)r)rrrr-)graphr/r4s r(rrxs+ uj) ==002C9J KKr*clt|ttj|j |yr,)r)rrr r-)sparser/s r(r r s(v01&224c:r*c^t|ttjfr,t j tj ||||yt|tr|n|j}tj|tkDr tdt j||||y)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_pathfspathbytesr-sys getsizeofr ValueErrorr)model full_checkskip_opset_compatibility_checkcheck_custom_domainprotobuf_strings r(rrs4%#r{{+,  IIe   *    u-E53J3J3L  == ),< <l    *   r*)r&rr'z type[Message]returnNone)r.rr/C.CheckerContextrWrX)r1rr/rYrWrX)r3rr/rYr4C.LexicalScopeContextrWrX)r6rr/rYr4rZrWrX)rBrr/zC.CheckerContext | Noner4rZrWrX)rErr/rYr4rZrWrX)rGrr/rYrWrX)FFF) rRz&ModelProto | str | bytes | os.PathLikerSboolrTr[rUr[rWrX)0__doc__ __future__r__all__rJrOtypingrrrgoogle.protobuf.messager onnx.defsr:onnx.onnx_cpp2py_export.checkeronnx_cpp2py_exportcheckerronnx.shape_inferencerrrrrrrrrrr9r r>defsonnx_opset_versionrALexicalScopeContextr rr)r r rrrrr rrr*r(rjs#K"   ))+++   #!""$'!#TYY%A%A%C D---/ :Xc3h%7 8 9HCC%5C C?N;,/DO O O-O O,/DJ J J-J J$(/DKK K-K K(,/DL L L-L L8G; ;$4; ;+0 % 0 10 0 %)0  0  0 f##r*