diff options
Diffstat (limited to 'init.c')
-rw-r--r-- | init.c | 5 |
1 files changed, 5 insertions, 0 deletions
@@ -95,6 +95,9 @@ #include "generic/VolumetricConvolution.c" #include "THGenerateFloatTypes.h" +#include "generic/VolumetricMaxPooling.c" +#include "THGenerateFloatTypes.h" + #include "generic/MultiMarginCriterion.c" #include "THGenerateFloatTypes.h" @@ -142,6 +145,7 @@ int luaopen_libnn(lua_State *L) nn_FloatSpatialSubSampling_init(L); nn_FloatSpatialMaxPooling_init(L); nn_FloatVolumetricConvolution_init(L); + nn_FloatVolumetricMaxPooling_init(L); nn_FloatMultiMarginCriterion_init(L); nn_FloatMultiLabelMarginCriterion_init(L); nn_FloatL1Cost_init(L); @@ -176,6 +180,7 @@ int luaopen_libnn(lua_State *L) nn_DoubleSpatialSubSampling_init(L); nn_DoubleSpatialMaxPooling_init(L); nn_DoubleVolumetricConvolution_init(L); + nn_DoubleVolumetricMaxPooling_init(L); nn_DoubleMultiMarginCriterion_init(L); nn_DoubleMultiLabelMarginCriterion_init(L); nn_DoubleL1Cost_init(L); |