diff options
| author | soumith <soumith@fb.com> | 2017-04-12 00:55:09 +0300 |
|---|---|---|
| committer | soumith <soumith@fb.com> | 2017-04-12 00:56:53 +0300 |
| commit | 89c27f5f1d3814c79119bb24760f1c15040fbfc5 (patch) | |
| tree | 84d129507ab7622b531b7e8c12d4e2dbbcc1ef65 | |
| parent | e0094cb05f43533f1ed6ff23a0575d62af2e827f (diff) | |
fix THNN headers
| -rw-r--r-- | lib/THNN/generic/THNN.h | 22 |
1 files changed, 11 insertions, 11 deletions
diff --git a/lib/THNN/generic/THNN.h b/lib/THNN/generic/THNN.h index 226526e..31849d6 100644 --- a/lib/THNN/generic/THNN.h +++ b/lib/THNN/generic/THNN.h @@ -457,11 +457,11 @@ TH_API void THNN_(SoftShrink_updateGradInput)( TH_API void THNN_(IndexLinear_updateOutput)( THNNState *state, - THLongTensor *keys, + THIndexTensor *keys, long keysOffset, THTensor *values, - THLongTensor *sizes, - THLongTensor *cumSumSizes, + THIndexTensor *sizes, + THIndexTensor *cumSumSizes, THTensor *output, THTensor *weight, THTensor *bias, @@ -469,11 +469,11 @@ TH_API void THNN_(IndexLinear_updateOutput)( int train); TH_API void THNN_(IndexLinear_accGradParameters)( THNNState *state, - THLongTensor *keys, + THIndexTensor *keys, long keysOffset, THTensor *values, - THLongTensor *sizes, - THLongTensor *cumSumSizes, + THIndexTensor *sizes, + THIndexTensor *cumSumSizes, THTensor *gradOutput, THTensor *gradWeight, THTensor *gradBias, @@ -484,11 +484,11 @@ TH_API void THNN_(IndexLinear_accGradParameters)( accreal scale); TH_API void THNN_(IndexLinear_accUpdateGradParameters)( THNNState *state, - THLongTensor *keys, + THIndexTensor *keys, long keysOffset, THTensor *values, - THLongTensor *sizes, - THLongTensor *cumSumSizes, + THIndexTensor *sizes, + THIndexTensor *cumSumSizes, THTensor *gradOutput, THTensor *weight, THTensor *bias, @@ -500,8 +500,8 @@ TH_API void THNN_(IndexLinear_updateParameters)( THTensor *gradBias, THTensor *weight, THTensor *bias, - THLongTensor *runningKeys, - THLongTensor *cumSumSizes, + THIndexTensor *runningKeys, + THIndexTensor *cumSumSizes, long keysOffset, accreal weightDecay, accreal learningRate); |