mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-30 03:34:56 +08:00
456c87c8a2
Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/139151 Approved by: https://github.com/ezyang Co-authored-by: Aaron Gokaslan <aaronGokaslan@gmail.com>
21 lines
505 B
C++
21 lines
505 B
C++
#pragma once
|
|
|
|
namespace torch::onnx {
|
|
|
|
enum class OperatorExportTypes {
|
|
ONNX, // Strict ONNX export
|
|
ONNX_ATEN, // ONNX With ATen op everywhere
|
|
ONNX_ATEN_FALLBACK, // ONNX export with ATen fallback
|
|
ONNX_FALLTHROUGH, // Export supported ONNX ops. Pass through unsupported ops.
|
|
};
|
|
|
|
enum class TrainingMode {
|
|
EVAL, // Inference mode
|
|
PRESERVE, // Preserve model state (eval/training)
|
|
TRAINING, // Training mode
|
|
};
|
|
|
|
constexpr auto kOnnxNodeNameAttribute = "onnx_name";
|
|
|
|
} // namespace torch::onnx
|