1 #include <torch/enum.h> 2 3 TORCH_ENUM_DEFINE(Linear) 4 TORCH_ENUM_DEFINE(Conv1D) 5 TORCH_ENUM_DEFINE(Conv2D) 6 TORCH_ENUM_DEFINE(Conv3D) 7 TORCH_ENUM_DEFINE(ConvTranspose1D) 8 TORCH_ENUM_DEFINE(ConvTranspose2D) 9 TORCH_ENUM_DEFINE(ConvTranspose3D) 10 TORCH_ENUM_DEFINE(Sigmoid) 11 TORCH_ENUM_DEFINE(Tanh) 12 TORCH_ENUM_DEFINE(ReLU) 13 TORCH_ENUM_DEFINE(GELU) 14 TORCH_ENUM_DEFINE(LeakyReLU) 15 TORCH_ENUM_DEFINE(FanIn) 16 TORCH_ENUM_DEFINE(FanOut) 17 TORCH_ENUM_DEFINE(Constant) 18 TORCH_ENUM_DEFINE(Reflect) 19 TORCH_ENUM_DEFINE(Replicate) 20 TORCH_ENUM_DEFINE(Circular) 21 TORCH_ENUM_DEFINE(Nearest) 22 TORCH_ENUM_DEFINE(Bilinear) 23 TORCH_ENUM_DEFINE(Bicubic) 24 TORCH_ENUM_DEFINE(Trilinear) 25 TORCH_ENUM_DEFINE(Area) 26 TORCH_ENUM_DEFINE(NearestExact) 27 TORCH_ENUM_DEFINE(Sum) 28 TORCH_ENUM_DEFINE(Mean) 29 TORCH_ENUM_DEFINE(Max) 30 TORCH_ENUM_DEFINE(None) 31 TORCH_ENUM_DEFINE(BatchMean) 32 TORCH_ENUM_DEFINE(Zeros) 33 TORCH_ENUM_DEFINE(Border) 34 TORCH_ENUM_DEFINE(Reflection) 35 TORCH_ENUM_DEFINE(RNN_TANH) 36 TORCH_ENUM_DEFINE(RNN_RELU) 37 TORCH_ENUM_DEFINE(LSTM) 38 TORCH_ENUM_DEFINE(GRU) 39 TORCH_ENUM_DEFINE(Valid) 40 TORCH_ENUM_DEFINE(Same) 41