Skip to content

Commit 352f8b2

Browse files
committed
Merge commit 'ced01f6c919c4b7109512ce797a2a0185c8f8112'
2 parents d351239 + ced01f6 commit 352f8b2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torch/lib/THCUNN/generic/THCUNN.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,8 @@ TH_API void THNN_(GRUFused_updateGradInput)(
183183
THCTensor *gradInInput,
184184
THCTensor *gradInHidden,
185185
THCTensor *gradOutput,
186-
THCTensor *gradInput);
186+
THCTensor *gradInputHx,
187+
THCTensor *storage);
187188

188189
TH_API void THNN_(LSTMFused_updateOutput)(
189190
THCState *state,

0 commit comments

Comments
 (0)