Caffe2 - C++ API
A deep learning, cross platform ML framework
modules.h
1 #pragma once
2 
3 #include <torch/nn/modules/any.h>
4 #include <torch/nn/modules/batchnorm.h>
5 #include <torch/nn/modules/conv.h>
6 #include <torch/nn/modules/dropout.h>
7 #include <torch/nn/modules/embedding.h>
8 #include <torch/nn/modules/functional.h>
9 #include <torch/nn/modules/linear.h>
10 #include <torch/nn/modules/rnn.h>
11 #include <torch/nn/modules/sequential.h>