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