#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
|
