added two generic upsamplers/resamplers modules

1 files changed, 10 insertions, 0 deletions

diff --git a/init.c b/init.c
index 1fdf687..50c5482 100644
--- a/init.c
+++ b/init.c
@@ -27,6 +27,12 @@ static const void* torch_DoubleTensor_id = NULL;
 #include "generic/SpatialLogSoftMax.c"
 #include "THGenerateFloatTypes.h"
 
+#include "generic/SpatialUpSampling.c"
+#include "THGenerateFloatTypes.h"
+
+#include "generic/SpatialReSampling.c"
+#include "THGenerateFloatTypes.h"
+
 #include "generic/Threshold.c"
 #include "THGenerateFloatTypes.h"
 
@@ -42,6 +48,8 @@ DLL_EXPORT int luaopen_libnnx(lua_State *L)
   nn_FloatSpatialConvolutionTable_init(L);
   nn_FloatSpatialLogSoftMax_init(L);
   nn_FloatSpatialMaxPooling_init(L);
+  nn_FloatSpatialUpSampling_init(L);
+  nn_FloatSpatialReSampling_init(L);
 
   nn_DoubleSpatialLinear_init(L);
   nn_DoubleHardShrink_init(L);
@@ -50,6 +58,8 @@ DLL_EXPORT int luaopen_libnnx(lua_State *L)
   nn_DoubleSpatialConvolutionTable_init(L);
   nn_DoubleSpatialLogSoftMax_init(L);
   nn_DoubleSpatialMaxPooling_init(L);
+  nn_DoubleSpatialUpSampling_init(L);
+  nn_DoubleSpatialReSampling_init(L);
 
   return 1;
 }