diff options
| author | soumith <soumith@fb.com> | 2016-02-25 04:11:21 +0300 |
|---|---|---|
| committer | soumith <soumith@fb.com> | 2016-02-25 04:23:10 +0300 |
| commit | 86ef4ae321db60435edf2d5d75979c99e01e1831 (patch) | |
| tree | 0b3f39712c891d3fc458482ea106f59fe7660f73 /THCUNN.lua | |
| parent | ef1f945f205e3615cdc2fd1f1d5cbe7e054fc6e9 (diff) | |
adding SpatialReflectionPadding and SpatialReplicationPadding
Diffstat (limited to 'THCUNN.lua')
| -rw-r--r-- | THCUNN.lua | 26 |
1 files changed, 26 insertions, 0 deletions
@@ -776,6 +776,32 @@ TH_API void THNN_CudaVolumetricMaxPooling_updateGradInput( THCudaTensor *indices, int dT, int dW, int dH, int padT, int padW, int padH); + +TH_API void THNN_CudaSpatialReflectionPadding_updateOutput(THCState *state, + THCudaTensor *input, + THCudaTensor *output, + int padL, int padR, + int padT, int padB + ); +TH_API void THNN_CudaSpatialReflectionPadding_updateGradInput(THCState *state, + THCudaTensor *input, + THCudaTensor *gradOutput, + THCudaTensor *gradInput, + int padL, int padR, + int padT, int padB); + +TH_API void THNN_CudaSpatialReplicationPadding_updateOutput(THCState *state, + THCudaTensor *input, + THCudaTensor *output, + int padL, int padR, + int padT, int padB + ); +TH_API void THNN_CudaSpatialReplicationPadding_updateGradInput(THCState *state, + THCudaTensor *input, + THCudaTensor *gradOutput, + THCudaTensor *gradInput, + int padL, int padR, + int padT, int padB); ]] local preprocessed = string.gsub(THCUNN_h, 'TH_API ', '') |