diff options
author | Soumith Chintala <soumith@gmail.com> | 2017-06-07 16:21:20 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2017-06-07 16:21:20 +0300 |
commit | a9c4d64850f2aecb1e29e0b42a3e801c13d192cd (patch) | |
tree | 3262e1dfc53b5db8d6c46ea245e69fb5e89564f9 | |
parent | 3087b8786be61c04270330b8fc2c628690ee0d0c (diff) |
fix GRUFused signature
-rw-r--r-- | lib/THCUNN/generic/THCUNN.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/lib/THCUNN/generic/THCUNN.h b/lib/THCUNN/generic/THCUNN.h index 6fc545c..4b7b64a 100644 --- a/lib/THCUNN/generic/THCUNN.h +++ b/lib/THCUNN/generic/THCUNN.h @@ -183,7 +183,8 @@ TH_API void THNN_(GRUFused_updateGradInput)( THCTensor *gradInInput, THCTensor *gradInHidden, THCTensor *gradOutput, - THCTensor *gradInput); + THCTensor *gradInputHx, + THCTensor *storage); TH_API void THNN_(LSTMFused_updateOutput)( THCState *state, |