Caffe2 - C++ API
A deep learning, cross platform ML framework
Tensor.cpp
1 #define __STDC_FORMAT_MACROS
2 
3 #include <torch/csrc/python_headers.h>
4 #include <structmember.h>
5 
6 #include <TH/THMath.h>
7 #include <vector>
8 #include <stack>
9 #include <tuple>
10 #include <torch/csrc/cuda/THCP.h>
11 
12 #include <torch/csrc/cuda/override_macros.h>
13 #include <torch/csrc/copy_utils.h>
14 #include <torch/csrc/DynamicTypes.h>
15 #include <torch/csrc/utils/tensor_numpy.h>
16 
17 //generic_include THC torch/csrc/generic/Tensor.cpp
18 
19 #include <torch/csrc/cuda/undef_macros.h>
20 #include <torch/csrc/cuda/restore_macros.h>