1#pragma once
2
3namespace torch {
4namespace onnx {
5
6enum class OperatorExportTypes {
7 ONNX, // Strict ONNX export
8 ONNX_ATEN, // ONNX With ATen op everywhere
9 ONNX_ATEN_FALLBACK, // ONNX export with ATen fallback
10 ONNX_FALLTHROUGH, // Export supported ONNX ops. Pass through unsupported ops.
11};
12
13enum class TrainingMode {
14 EVAL, // Inference mode
15 PRESERVE, // Preserve model state (eval/training)
16 TRAINING, // Training mode
17};
18
19constexpr char kOnnxNodeNameAttribute[] = "onnx_name";
20
21} // namespace onnx
22} // namespace torch
23