Caffe2 - C++ API
A deep learning, cross platform ML framework
Storage.cpp
1 #define __STDC_FORMAT_MACROS
2 
3 #include <torch/csrc/python_headers.h>
4 #include <structmember.h>
5 
6 // See Note [TH abstraction violation]
7 // - Used to get at allocator from storage
8 #include <TH/THTensor.hpp>
9 #include <THC/THCTensor.hpp>
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 
16 #define THC_GENERIC_FILE "torch/csrc/generic/Storage.cpp"
17 #include <THC/THCGenerateAllTypes.h>
18 
19 #define THC_GENERIC_FILE "torch/csrc/generic/Storage.cpp"
20 #include <THC/THCGenerateBoolType.h>