blob: b2a528ec36b040c42444d7a56249c3c918e44c96 (
plain)
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
|
#include "TH.h"
#include "luaT.h"
#define torch_(NAME) TH_CONCAT_3(torch_, Real, NAME)
#define torch_string_(NAME) TH_CONCAT_STRING_3(torch., Real, NAME)
#define nn_(NAME) TH_CONCAT_3(nn_, Real, NAME)
static const void* torch_FloatTensor_id = NULL;
static const void* torch_DoubleTensor_id = NULL;
#include "generic/Square.c"
#include "THGenerateFloatTypes.h"
#include "generic/Sqrt.c"
#include "THGenerateFloatTypes.h"
#include "generic/HardTanh.c"
#include "THGenerateFloatTypes.h"
#include "generic/Exp.c"
#include "THGenerateFloatTypes.h"
#include "generic/LogSigmoid.c"
#include "THGenerateFloatTypes.h"
#include "generic/LogSoftMax.c"
#include "THGenerateFloatTypes.h"
#include "generic/Sigmoid.c"
#include "THGenerateFloatTypes.h"
#include "generic/SoftPlus.c"
#include "THGenerateFloatTypes.h"
#include "generic/Tanh.c"
#include "THGenerateFloatTypes.h"
#include "generic/Abs.c"
#include "THGenerateFloatTypes.h"
#include "generic/HardShrink.c"
#include "THGenerateFloatTypes.h"
#include "generic/SoftShrink.c"
#include "THGenerateFloatTypes.h"
#include "generic/Threshold.c"
#include "THGenerateFloatTypes.h"
#include "generic/SoftMax.c"
#include "THGenerateFloatTypes.h"
#include "generic/Max.c"
#include "THGenerateFloatTypes.h"
#include "generic/Min.c"
#include "THGenerateFloatTypes.h"
#include "generic/MSECriterion.c"
#include "THGenerateFloatTypes.h"
#include "generic/AbsCriterion.c"
#include "THGenerateFloatTypes.h"
#include "generic/SparseLinear.c"
#include "THGenerateFloatTypes.h"
#include "generic/TemporalConvolution.c"
#include "THGenerateFloatTypes.h"
#include "generic/TemporalSubSampling.c"
#include "THGenerateFloatTypes.h"
#include "generic/SpatialConvolution.c"
#include "THGenerateFloatTypes.h"
#include "generic/SpatialConvolutionMap.c"
#include "THGenerateFloatTypes.h"
#include "generic/SpatialSubSampling.c"
#include "THGenerateFloatTypes.h"
#include "generic/SpatialMaxPooling.c"
#include "THGenerateFloatTypes.h"
#include "generic/VolumetricConvolution.c"
#include "THGenerateFloatTypes.h"
#include "generic/MultiMarginCriterion.c"
#include "THGenerateFloatTypes.h"
#include "generic/MultiLabelMarginCriterion.c"
#include "THGenerateFloatTypes.h"
DLL_EXPORT int luaopen_libnn(lua_State *L)
{
torch_FloatTensor_id = luaT_checktypename2id(L, "torch.FloatTensor");
torch_DoubleTensor_id = luaT_checktypename2id(L, "torch.DoubleTensor");
lua_newtable(L);
lua_pushvalue(L, -1);
lua_setfield(L, LUA_GLOBALSINDEX, "nn");
nn_FloatMin_init(L);
nn_FloatMax_init(L);
nn_FloatExp_init(L);
nn_FloatSqrt_init(L);
nn_FloatSquare_init(L);
nn_FloatHardTanh_init(L);
nn_FloatLogSoftMax_init(L);
nn_FloatMSECriterion_init(L);
nn_FloatAbsCriterion_init(L);
nn_FloatLogSigmoid_init(L);
nn_FloatSigmoid_init(L);
nn_FloatSoftMax_init(L);
nn_FloatSoftPlus_init(L);
nn_FloatTanh_init(L);
nn_FloatAbs_init(L);
nn_FloatHardShrink_init(L);
nn_FloatSoftShrink_init(L);
nn_FloatThreshold_init(L);
nn_FloatSparseLinear_init(L);
nn_FloatTemporalConvolution_init(L);
nn_FloatTemporalSubSampling_init(L);
nn_FloatSpatialConvolution_init(L);
nn_FloatSpatialConvolutionMap_init(L);
nn_FloatSpatialSubSampling_init(L);
nn_FloatSpatialMaxPooling_init(L);
nn_FloatVolumetricConvolution_init(L);
nn_FloatMultiMarginCriterion_init(L);
nn_FloatMultiLabelMarginCriterion_init(L);
nn_DoubleMin_init(L);
nn_DoubleMax_init(L);
nn_DoubleExp_init(L);
nn_DoubleSqrt_init(L);
nn_DoubleSquare_init(L);
nn_DoubleHardTanh_init(L);
nn_DoubleLogSoftMax_init(L);
nn_DoubleMSECriterion_init(L);
nn_DoubleAbsCriterion_init(L);
nn_DoubleLogSigmoid_init(L);
nn_DoubleSigmoid_init(L);
nn_DoubleSoftMax_init(L);
nn_DoubleSoftPlus_init(L);
nn_DoubleTanh_init(L);
nn_DoubleAbs_init(L);
nn_DoubleHardShrink_init(L);
nn_DoubleSoftShrink_init(L);
nn_DoubleThreshold_init(L);
nn_DoubleSparseLinear_init(L);
nn_DoubleTemporalConvolution_init(L);
nn_DoubleTemporalSubSampling_init(L);
nn_DoubleSpatialConvolution_init(L);
nn_DoubleSpatialConvolutionMap_init(L);
nn_DoubleSpatialSubSampling_init(L);
nn_DoubleSpatialMaxPooling_init(L);
nn_DoubleVolumetricConvolution_init(L);
nn_DoubleMultiMarginCriterion_init(L);
nn_DoubleMultiLabelMarginCriterion_init(L);
return 1;
}
|