U <cc @sFdZddlZddlmmZddlmZmZGdddZ e Z dS)zGlobals used internally by the ONNX exporter. Do not use this module outside of `torch.onnx` and its tests. Be very judicious when adding any new global variables. Do not create new global variables unless they are absolutely necessary. N) _constants_exporter_statesc@seZdZdZddZeddZejej dddZee dd d Z e je d d d Z ee dd dZ e je d ddZ dS)_InternalGlobalszGlobals used internally by ONNX exporter. NOTE: Be very judicious when adding any new variables. Do not create new global variables unless they are absolutely necessary. cCsptj|_tjj|_d|_d|_tj j |_ d|_ t ddkrHtjj|_n$t ddkrbtjj|_n tjj|_dS)NFTZ*TORCH_ONNX_EXPERIMENTAL_RUNTIME_TYPE_CHECKERRORSDISABLED)rZONNX_DEFAULT_OPSET_export_onnx_opset_version_C_onnx TrainingModeZEVAL_training_mode_in_onnx_exportZexport_trainingZOperatorExportTypesZONNXZoperator_export_typeZonnx_shape_inferenceosgetenvrZRuntimeTypeCheckStaterZruntime_type_check_staterZWARNINGSselfr7/tmp/pip-unpacked-wheel-gikjz4vx/torch/onnx/_globals.py__init__s z_InternalGlobals.__init__cCs|jS)z#The training mode for the exporter.)r rrrr training_mode1sz_InternalGlobals.training_mode)rcCst|tjstd||_dS)Nz\training_mode must be of type 'torch.onnx.TrainingMode'. This is likely a bug in torch.onnx.) isinstancerr TypeErrorr )rrrrrr6s  )returncCs|jS)z!Opset version used during export.)rrrrrexport_onnx_opset_version?sz*_InternalGlobals.export_onnx_opset_version)valuecCs2ttjtjd}||kr(td|||_dS)Nz Unsupported ONNX opset version: )rangerZONNX_MIN_OPSETZONNX_MAX_OPSET ValueErrorr)rrZsupported_versionsrrrrDscCs|jS)z+Whether it is in the middle of ONNX export.)r rrrrin_onnx_exportMsz_InternalGlobals.in_onnx_exportcCst|tk rtd||_dS)Nz in_onnx_export must be a boolean)typeboolrr )rrrrrrRs N)__name__ __module__ __qualname____doc__rpropertyrsetterrr intrrrrrrrrs r) r"r Ztorch._C._onnxZ_CZ_onnxrZ torch.onnxrrrZGLOBALSrrrrs H