1 | #pragma once |
2 | |
3 | namespace torch { |
4 | namespace onnx { |
5 | |
6 | enum 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 | |
13 | enum class TrainingMode { |
14 | EVAL, // Inference mode |
15 | PRESERVE, // Preserve model state (eval/training) |
16 | TRAINING, // Training mode |
17 | }; |
18 | |
19 | constexpr char kOnnxNodeNameAttribute[] = "onnx_name" ; |
20 | |
21 | } // namespace onnx |
22 | } // namespace torch |
23 | |