diff options
| author | soumith <soumith@fb.com> | 2016-02-25 03:59:30 +0300 |
|---|---|---|
| committer | soumith <soumith@fb.com> | 2016-02-25 03:59:30 +0300 |
| commit | 47f58cff33cc86e52b72b02c468a72d810e4216d (patch) | |
| tree | a47da6251611d931d9465166755aa925f117b2bd /THNN.lua | |
| parent | bcdda102686713037538d770ee5c033879450ee3 (diff) | |
Adding SpatialReflection and SpatialReplication padding
Diffstat (limited to 'THNN.lua')
| -rw-r--r-- | THNN.lua | 26 |
1 files changed, 26 insertions, 0 deletions
@@ -947,6 +947,32 @@ TH_API void THNN_(VolumetricMaxUnpooling_updateGradInput)( int dT, int dW, int dH, int pT, int pW, int pH); +TH_API void THNN_(SpatialReflectionPadding_updateOutput)(THNNState *state, + THTensor *input, + THTensor *output, + int pad_l, int pad_r, + int pad_t, int pad_b); + +TH_API void THNN_(SpatialReflectionPadding_updateGradInput)(THNNState *state, + THTensor *input, + THTensor *gradOutput, + THTensor *gradInput, + int pad_l, int pad_r, + int pad_t, int pad_b); + +TH_API void THNN_(SpatialReplicationPadding_updateOutput)(THNNState *state, + THTensor *input, + THTensor *output, + int pad_l, int pad_r, + int pad_t, int pad_b); + +TH_API void THNN_(SpatialReplicationPadding_updateGradInput)(THNNState *state, + THTensor *input, + THTensor *gradOutput, + THTensor *gradInput, + int pad_l, int pad_r, + int pad_t, int pad_b); + ]] -- THGenerator struct declaration copied from torch7/lib/TH/THRandom.h |