encoding_lib.cpp 1 KB
Newer Older
Hang Zhang's avatar
init  
Hang Zhang committed
1
2
3
4
5
6
7
8
9
10
11
/*+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 * Created by: Hang Zhang
 * ECE Department, Rutgers University
 * Email: zhang.hang@rutgers.edu
 * Copyright (c) 2017
 *
 * This source code is licensed under the MIT-style license found in the
 * LICENSE file in the root directory of this source tree 
 *+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 */
#include <THC/THC.h>
Hang Zhang's avatar
Hang Zhang committed
12
#include <thc_encoding.h>
Hang Zhang's avatar
init  
Hang Zhang committed
13
14
15

extern THCState *state;

Hang Zhang's avatar
tested  
Hang Zhang committed
16
17
18
19
#ifdef __cplusplus
extern "C" {
#endif

Hang Zhang's avatar
v1.0.1  
Hang Zhang committed
20
// float
Hang Zhang's avatar
init  
Hang Zhang committed
21
22
#include "generic/encoding_generic.c"
#include "THC/THCGenerateFloatType.h"
Hang Zhang's avatar
tested  
Hang Zhang committed
23

Hang Zhang's avatar
v1.0.1  
Hang Zhang committed
24
25
26
27
28
29
30
#include "generic/syncbn_generic.c"
#include "THC/THCGenerateFloatType.h"

#include "generic/pooling_generic.c"
#include "THC/THCGenerateFloatType.h"

// double
Hang Zhang's avatar
Hang Zhang committed
31
32
33
#include "generic/encoding_generic.c"
#include "THC/THCGenerateDoubleType.h"

Hang Zhang's avatar
v1.0.1  
Hang Zhang committed
34
35
36
37
38
39
#include "generic/syncbn_generic.c"
#include "THC/THCGenerateDoubleType.h"

#include "generic/pooling_generic.c"
#include "THC/THCGenerateDoubleType.h"

Hang Zhang's avatar
tested  
Hang Zhang committed
40
41
42
#ifdef __cplusplus
}
#endif