U <ºc¡ã@sdddlmZddlZddlmZddlmZGdd„dƒZGdd„dƒZej Gd d „d ej ƒƒZ dS) é)Ú annotationsN)ÚDict)Ú_Cc@s eZdZdZdZdZdZdZdS)Ú ExportTypesz'Specifies how the ONNX model is stored.z+Saves model in the specified protobuf file.z5Saves model in the specified ZIP file (uncompressed).z3Saves model in the specified ZIP file (compressed).z$Saves model in the specified folder.N)Ú__name__Ú __module__Ú __qualname__Ú__doc__Z PROTOBUF_FILEZ ZIP_ARCHIVEZCOMPRESSED_ZIP_ARCHIVEZ DIRECTORY©r r ú?/tmp/pip-unpacked-wheel-gikjz4vx/torch/onnx/_exporter_states.pyr s rc@s$eZdZdZdddddœdd„Zd S) ÚSymbolicContexta¡Extra context for symbolic functions. Args: params_dict (Dict[str, _C.IValue]): Mapping from graph initializer name to IValue. env (Dict[_C.Value, _C.Value]): Mapping from Torch domain graph Value to ONNX domain graph Value. cur_node (_C.Node): Current node being converted to ONNX domain. onnx_block (_C.Block): Current ONNX block that converted nodes are being appended to. zDict[str, _C.IValue]Údictz_C.Nodez_C.Block©Ú params_dictÚenvÚcur_nodeÚ onnx_blockcCs||_||_||_||_dS)Nr)Úselfrrrrr r r Ú__init__szSymbolicContext.__init__N)rrrr rr r r r r s r c@s(eZdZdZe ¡Ze ¡Ze ¡ZdS)ÚRuntimeTypeCheckStatezRuntime type check state.N) rrrr ÚenumÚautoZDISABLEDZWARNINGSZERRORSr r r r r+sr) Ú __future__rrÚtypingrZtorchrrr ÚuniqueÚEnumrr r r r Ús