Caffe2 - C++ API
A deep learning, cross platform ML framework
serialization.cpp
1 #include <torch/csrc/python_headers.h>
2 
3 #include <torch/csrc/cuda/THCP.h>
4 
5 #include <torch/csrc/cuda/override_macros.h>
6 
7 #include <system_error>
8 #include <memory>
9 
10 #define THC_GENERIC_FILE "torch/csrc/generic/serialization.cpp"
11 #include <THC/THCGenerateAllTypes.h>
12 
13 #define THC_GENERIC_FILE "torch/csrc/generic/serialization.cpp"
14 #include <THC/THCGenerateBoolType.h>