35 строки
1.1 KiB
Python
35 строки
1.1 KiB
Python
# -------------------------------------------------------------------------
|
|
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
# --------------------------------------------------------------------------
|
|
from olive.common.utils import StrEnumBase
|
|
|
|
|
|
class Framework(StrEnumBase):
|
|
"""Framework of the model."""
|
|
|
|
ONNX = "ONNX"
|
|
PYTORCH = "PyTorch"
|
|
QNN = "QNN"
|
|
TENSORFLOW = "TensorFlow"
|
|
SNPE = "SNPE"
|
|
OPENVINO = "OpenVINO"
|
|
|
|
|
|
class ModelFileFormat(StrEnumBase):
|
|
"""Given a framework, there might be 1 or more on-disk model file format(s), model save/Load logic may differ."""
|
|
|
|
ONNX = "ONNX"
|
|
PYTORCH_ENTIRE_MODEL = "PyTorch.EntireModel"
|
|
PYTORCH_STATE_DICT = "PyTorch.StateDict"
|
|
PYTORCH_TORCH_SCRIPT = "PyTorch.TorchScript"
|
|
PYTORCH_SLICE_GPT_MODEL = "PyTorch.SliceGPT"
|
|
TENSORFLOW_PROTOBUF = "TensorFlow.Protobuf"
|
|
TENSORFLOW_SAVED_MODEL = "TensorFlow.SavedModel"
|
|
SNPE_DLC = "SNPE.DLC"
|
|
QNN_CPP = "QNN.CPP"
|
|
QNN_LIB = "QNN.LIB"
|
|
QNN_SERIALIZED_BIN = "QNN.SERIALIZED.BIN"
|
|
OPENVINO_IR = "OpenVINO.IR"
|
|
COMPOSITE_MODEL = "Composite"
|