/
usr
/
local
/
lib64
/
python3.6
/
site-packages
/
torch
/
include
/
torch
/
csrc
/
onnx
/
/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/onnx
mkdir
upload
Name
Size
Mode
Actions
init.h
158
0644
edit
dl
rm
onnx.h
699
0644
edit
dl
rm
Edit:
/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/onnx/onnx.h
(699B)
#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 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 instead of using onnx::IR_VERSION so that the // test_operators.py will be more stable. Only bump it when // necessary. static const size_t IR_VERSION = 7; static const char* PRODUCER_VERSION = "1.10"; } // namespace onnx } // namespace torch
Save
cmd:
run