| #ifndef THNN_H | |
| #define THNN_H | |
| #include <stdbool.h> | |
| #include <TH.h> | |
| #ifdef _OPENMP | |
| #include <omp.h> | |
| #endif | |
| #define THNN_(NAME) TH_CONCAT_3(THNN_, Real, NAME) | |
| #define THIndexTensor THLongTensor | |
| #define THIndexTensor_(NAME) THLongTensor_ ## NAME | |
| #define THIntegerTensor THIntTensor | |
| #define THIntegerTensor_(NAME) THIntTensor_ ## NAME | |
| typedef long THIndex_t; | |
| typedef int THInteger_t; | |
| typedef void THNNState; | |
| #include "generic/THNN.h" | |
| #include <THGenerateFloatTypes.h> | |
| #endif |