xref: /aosp_15_r20/external/pytorch/torch/onnx/_exporter_states.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1from __future__ import annotations
2
3
4class ExportTypes:
5    """Specifies how the ONNX model is stored."""
6
7    # TODO(justinchuby): Deprecate and remove this class.
8
9    PROTOBUF_FILE = "Saves model in the specified protobuf file."
10    ZIP_ARCHIVE = "Saves model in the specified ZIP file (uncompressed)."
11    COMPRESSED_ZIP_ARCHIVE = "Saves model in the specified ZIP file (compressed)."
12    DIRECTORY = "Saves model in the specified folder."
13