diff options
| author | Frederic Besse <fbesse@google.com> | 2016-02-18 18:33:00 +0300 |
|---|---|---|
| committer | Frederic Besse <fbesse@google.com> | 2016-02-18 18:33:00 +0300 |
| commit | 0e238437ad51850eedc453db31bea050b56f6bd4 (patch) | |
| tree | a30fb8a0208bb5f25599dc2bf7393895b77ae219 /THCUNN.lua | |
| parent | c837a7ff03c3c624c3230227b304d48e8b06cb52 (diff) | |
Reimplemented VolumetricFullConvolution in the same fashion as SpatialFullConvolution.
Diffstat (limited to 'THCUNN.lua')
| -rw-r--r-- | THCUNN.lua | 9 |
1 files changed, 6 insertions, 3 deletions
@@ -730,7 +730,8 @@ TH_API void THNN_CudaVolumetricFullConvolution_updateOutput( THCudaTensor *finput, THCudaTensor *fgradInput, int dT, int dW, int dH, - int pT, int pW, int pH); + int padT, int padW, int padH, + int adjT, int adjW, int adjH); TH_API void THNN_CudaVolumetricFullConvolution_updateGradInput( THCState *state, THCudaTensor *input, @@ -740,7 +741,8 @@ TH_API void THNN_CudaVolumetricFullConvolution_updateGradInput( THCudaTensor *finput, THCudaTensor *fgradInput, int dT, int dW, int dH, - int pT, int pW, int pH); + int padT, int padW, int padH, + int adjT, int adjW, int adjH); TH_API void THNN_CudaVolumetricFullConvolution_accGradParameters( THCState *state, THCudaTensor *input, @@ -750,7 +752,8 @@ TH_API void THNN_CudaVolumetricFullConvolution_accGradParameters( THCudaTensor *finput, THCudaTensor *fgradInput, int dT, int dW, int dH, - int pT, int pW, int pH, + int padT, int padW, int padH, + int adjT, int adjW, int adjH, float scale); TH_API void THNN_CudaVolumetricMaxPooling_updateOutput( |