U MhU@sDddlmZddlmZddlmZGdddZGdddZdS) ) annotations)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 DIRECTORYr r M/var/www/html/venv/lib/python3.8/site-packages/torch/onnx/_exporter_states.pyrs rc@s$eZdZdZdddddddZd S) SymbolicContextaExtra 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_dictenvcur_node onnx_blockcCs||_||_||_||_dS)Nr)selfrrrrr r r __init__szSymbolicContext.__init__N)rrrr rr r r r r s r N) __future__rtypingrZtorchrrr r r r r s