diff options
Diffstat (limited to 'lib/THCUNN/generic/THCUNN.h')
| -rw-r--r-- | lib/THCUNN/generic/THCUNN.h | 38 |
1 files changed, 38 insertions, 0 deletions
diff --git a/lib/THCUNN/generic/THCUNN.h b/lib/THCUNN/generic/THCUNN.h index 79426b7..b44fff3 100644 --- a/lib/THCUNN/generic/THCUNN.h +++ b/lib/THCUNN/generic/THCUNN.h @@ -616,6 +616,44 @@ TH_API void THNN_(SpatialConvolutionMM_accGradParameters)( int padW, int padH, accreal scale); +TH_API void THNN_(SpatialDepthWiseConvolution_updateOutput)( + THCState *state, + THCTensor *input, + THCTensor *output, + THCTensor *weight, + THCTensor *bias, // [OPTIONAL] + THCTensor *columns, + THCTensor *ones, + int kW, int kH, + int dW, int dH, + int padW, int padH); + +TH_API void THNN_(SpatialDepthWiseConvolution_updateGradInput)( + THCState *state, + THCTensor *input, + THCTensor *gradOutput, + THCTensor *gradInput, + THCTensor *weight, + THCTensor *gradColumns, + THCTensor *ones, + int kW, int kH, + int dW, int dH, + int padW, int padH); + +TH_API void THNN_(SpatialDepthWiseConvolution_accGradParameters)( + THCState *state, + THCTensor *input, + THCTensor *gradOutput, + THCTensor *gradWeight, + THCTensor *gradBias, // [OPTIONAL] + THCTensor *columns, + THCTensor *ones, + int kW, int kH, + int dW, int dH, + int padW, int padH, + accreal scale); + + TH_API void THNN_(SpatialCrossMapLRN_updateOutput)( THCState *state, THCTensor *input, |