forked from torch/cunn
-
Notifications
You must be signed in to change notification settings - Fork 1
/
utils.h
46 lines (41 loc) · 1.51 KB
/
utils.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
#ifndef CUNN_UTILS_H
#define CUNN_UTILS_H
extern "C"
{
#include <lua.h>
}
#include <luaT.h>
#include <THC/THC.h>
THCState* getCutorchState(lua_State* L);
void cunn_ClassNLLCriterion_init(lua_State *L);
void cunn_Tanh_init(lua_State *L);
void cunn_Sigmoid_init(lua_State *L);
void cunn_HardTanh_init(lua_State *L);
void cunn_L1Cost_init(lua_State *L);
void cunn_LogSoftMax_init(lua_State *L);
void cunn_SoftMax_init(lua_State *L);
void cunn_TemporalConvolution_init(lua_State *L);
void cunn_TemporalMaxPooling_init(lua_State *L);
void cunn_SpatialConvolutionMM_init(lua_State *L);
void cunn_SpatialMaxPooling_init(lua_State *L);
void cunn_SpatialAdaptiveMaxPooling_init(lua_State *L);
void cunn_SpatialSubSampling_init(lua_State *L);
void cunn_SpatialAveragePooling_init(lua_State *L);
void cunn_MultiMarginCriterion_init(lua_State *L);
void cunn_Square_init(lua_State *L);
void cunn_Sqrt_init(lua_State *L);
void cunn_Threshold_init(lua_State *L);
void cunn_MSECriterion_init(lua_State *L);
void cunn_SmoothL1Criterion_init(lua_State *L);
void cunn_AbsCriterion_init(lua_State *L);
void cunn_DistKLDivCriterion_init(lua_State *L);
void cunn_Abs_init(lua_State *L);
void cunn_SoftPlus_init(lua_State *L);
void cunn_SpatialUpSamplingNearest_init(lua_State *L);
void cunn_VolumetricConvolution_init(lua_State *L);
void cunn_VolumetricMaxPooling_init(lua_State *L);
void cunn_VolumetricAveragePooling_init(lua_State *L);
void cunn_LogSigmoid_init(lua_State *L);
void cunn_PReLU_init(lua_State *L);
void cunn_LookupTable_init(lua_State *L);
#endif