blob: 76170e18110f1be9d75ee7df31f930ef99704cdd [file] [log] [blame]
#pragma once
namespace torch { namespace onnx {
enum class OperatorExportTypes {
ONNX, // Strict ONNX export
ONNX_ATEN, // ONNX With ATen op everywhere
ONNX_ATEN_FALLBACK, // ONNX export with ATen fallback
RAW, // Raw export (no ONNX)
};
}} // namespace torch::onnx