1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
#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)
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
};
// we pin IR version to version 6 (12/11/2019) instead of using
// onnx::IR_VERSION. with this change, the test_operators.py will be more
// stable. only bump it when it's necessary
static const size_t IR_VERSION = 6;
static const char* PRODUCER_VERSION = "1.7";
}} // namespace torch::onnx
|
