Caffe2 - C++ API
A deep learning, cross platform ML framework
nn.h
1 #pragma once
2 
3 #include <torch/nn/cloneable.h>
4 #include <torch/nn/init.h>
5 #include <torch/nn/module.h>
6 #include <torch/nn/modules.h>
7 #include <torch/nn/pimpl.h>