U qh @sdZddlmZddddddd d d d d ddg ZddlZddlZddlmZmZm Z ddl m Z ddl Z ddlmmZddlZ ddl mZmZmZmZmZmZmZmZmZdZeZee_de j !ie_"e#Z$e dedefdZ%ddddddZ&efdd dd!d"d Z'efd#d dd$d%d Z(ee$fd&d d'dd(d)dZ)ee$fd*d d'dd+d,dZ*de$fd-d.d'dd/d0dZ+ee$fd1d d'dd2d3dZ,efd4d dd5d6dZ-dqsz"check_function..) r*rrCheckerContextonnxhelperZfind_min_ir_version_forZ opset_import ir_version opset_importsrr.)r3r,r1r(r(r)rfs r)graphr,r1r#cCst|tt|||Sr-)r*rrrr.)r=r,r1r(r(r)rxs r)sparser,r#cCst|tt||dSr-)r*rrrr.)r>r,r(r(r)rs Fz&ModelProto | str | bytes | os.PathLikebool)model full_checkskip_opset_compatibility_checkcheck_custom_domainr#cCsht|ttjfr(tt||||n= 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%strosPathLikerZcheck_model_pathfspathbytesr.sys getsizeofr ValueErrorr)r@rArBrCZprotobuf_stringr(r(r)rs&)FFF)0__doc__ __future__r__all__rErItypingrrrZgoogle.protobuf.messagerZ onnx.defsr9Zonnx.onnx_cpp2py_export.checkerZonnx_cpp2py_exportcheckerrZonnx.shape_inferencerrrrrrrrrrr8r r;ZdefsZonnx_opset_versionr<ZLexicalScopeContextr rr*r r rrrrrrr r(r(r(r)sh  ,     3