diff options
Diffstat (limited to 'lib/THNN/generic/THNN.h')
-rw-r--r-- | lib/THNN/generic/THNN.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/lib/THNN/generic/THNN.h b/lib/THNN/generic/THNN.h index 76a28eb..31b0795 100644 --- a/lib/THNN/generic/THNN.h +++ b/lib/THNN/generic/THNN.h @@ -1171,6 +1171,18 @@ TH_API void THNN_(SpatialUpSamplingBilinear_updateGradInput)( int outputHeight, int outputWidth); +TH_API void THNN_(SpatialGridSamplerBilinear_updateOutput)( + THNNState *state, + THTensor *input, + THTensor *grid, + THTensor *output); + +TH_API void THNN_(SpatialGridSamplerBilinear_updateGradInput)( + THNNState *state, + THTensor *input, THTensor *gradInput, + THTensor *grid, THTensor *gradGrid, + THTensor *gradOutput); + TH_API void THNN_(unfolded_acc)( THTensor *finput, THTensor *input, |