diff options
| author | soumith <soumith@fb.com> | 2017-04-12 00:55:09 +0300 |
|---|---|---|
| committer | soumith <soumith@fb.com> | 2017-04-12 00:57:11 +0300 |
| commit | e2469527b1f85802e93ace4ce709bdabd7f1010a (patch) | |
| tree | 35f405d242c345706c964b2bfc1f3973cacbf4ba | |
| parent | ca8fbb72bb6709f518f3ca8807474595db0690ad (diff) | |
fix THNN headers
| -rw-r--r-- | lib/THCUNN/generic/THCUNN.h | 22 |
1 files changed, 11 insertions, 11 deletions
diff --git a/lib/THCUNN/generic/THCUNN.h b/lib/THCUNN/generic/THCUNN.h index 4ab2949..da8e41e 100644 --- a/lib/THCUNN/generic/THCUNN.h +++ b/lib/THCUNN/generic/THCUNN.h @@ -417,11 +417,11 @@ TH_API void THNN_(SparseLinear_updateParameters)( TH_API void THNN_(IndexLinear_updateOutput)( THCState *state, - THCudaLongTensor *keys, + THCIndexTensor *keys, long keysOffset, THCTensor *values, - THCudaLongTensor *sizes, - THCudaLongTensor *cumSumSizes, + THCIndexTensor *sizes, + THCIndexTensor *cumSumSizes, THCTensor *output, THCTensor *weight, THCTensor *bias, @@ -430,11 +430,11 @@ TH_API void THNN_(IndexLinear_updateOutput)( TH_API void THNN_(IndexLinear_accGradParameters)( THCState *state, - THCudaLongTensor *keys, + THCIndexTensor *keys, long keysOffset, THCTensor *values, - THCudaLongTensor *sizes, - THCudaLongTensor *cumSumSizes, + THCIndexTensor *sizes, + THCIndexTensor *cumSumSizes, THCTensor *gradOutput, THCTensor *gradWeight, THCTensor *gradBias, @@ -446,11 +446,11 @@ TH_API void THNN_(IndexLinear_accGradParameters)( TH_API void THNN_(IndexLinear_accUpdateGradParameters)( THCState *state, - THCudaLongTensor *keys, + THCIndexTensor *keys, long keysOffset, THCTensor *values, - THCudaLongTensor *sizes, - THCudaLongTensor *cumSumSizes, + THCIndexTensor *sizes, + THCIndexTensor *cumSumSizes, THCTensor *gradOutput, THCTensor *weight, THCTensor *bias, @@ -463,8 +463,8 @@ TH_API void THNN_(IndexLinear_updateParameters)( THCTensor *gradBias, THCTensor *weight, THCTensor *bias, - THCudaLongTensor *runningKeys, - THCudaLongTensor *cumSumSizes, + THCIndexTensor *runningKeys, + THCIndexTensor *cumSumSizes, long keysOffset, accreal weightDecay, accreal learningRate); |