diff options
-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, |