o wi@sldZddlmZgdZddlZddlZddlmZmZm Z m Z ddl Z ddl mmZddlZ ddl mZmZmZmZmZmZmZmZmZerOddlmZdZeZee_ d e j!"ie_#e$Z%e d e d efd Z&d?ddZ'efd@ddZ(efdAddZ)ee%fdBd#d$Z*ee%fdCd'd(Z+de%fdDd,d-Z,ee%fdEd0d1Z-efdFd4d5Z. 6 6 6dGdHd=d>Z/ej0Z0dS)IzDGraph 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_CHECKINGAnyCallableTypeVar) IR_VERSIONAttributeProto FunctionProto GraphProto ModelProto NodeProtoSparseTensorProto TensorProtoValueInfoProto)MessagelFuncType.)boundprotor proto_type type[Message]returnNonecCs t||std|jddS)Nz'The proto message needs to be of type '') isinstance TypeError__name__)r!r"r*I/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/checker.py_ensure_proto_typeBs  r, value_inforctxC.CheckerContextcCt|tt||SN)r,rrr SerializeToString)r-r.r*r*r+r Is r tensorrcCr0r1)r,rrr r2)r3r.r*r*r+r Ps r attrrlexical_scope_ctxC.LexicalScopeContextcCt|tt|||Sr1)r,rrrr2)r4r.r5r*r*r+rU rnodercCr7r1)r,rrrr2)r9r.r5r*r*r+r^r8rfunctionrC.CheckerContext | NonecCsVt|t|dur t}tjj|jdd|_dd|jD|_ t | ||dS)NT)ignore_unknowncSsi|]}|j|jqSr*)domainversion).0domain_versionr*r*r+ rsz"check_function..) r,rrCheckerContextonnxhelperfind_min_ir_version_for opset_import ir_version opset_importsrr2)r:r.r5r*r*r+rgs rgraphrcCr7r1)r,rrrr2)rIr.r5r*r*r+ryr8rsparsercCst|tt||dSr1)r,rrrr2)rJr.r*r*r+rs rFmodel&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)r'strosPathLikercheck_model_pathfspathbytesr2sys getsizeofr ValueErrorr)rKrMrOrPprotobuf_stringr*r*r+rs&r)r!rr"r#r$r%)r-rr.r/r$r%)r3rr.r/r$r%)r4rr.r/r5r6r$r%)r9rr.r/r5r6r$r%)r:rr.r;r5r6r$r%)rIrr.r/r5r6r$r%)rJrr.r/r$r%)FFF) rKrLrMrNrOrNrPrNr$r%)1__doc__ __future__r__all__rRrWtypingrrrr onnx.defsrConnx.onnx_cpp2py_export.checkeronnx_cpp2py_exportcheckerronnx.shape_inferencerrrrrrrrrgoogle.protobuf.messagerrrBr rGdefsonnx_opset_versionrHLexicalScopeContextr rr,r r rrrrrrr r*r*r*r+sR ,            3