fix GRUFused signature

soumith · soumith · commit a9c4d64850f2 · 2017-06-07T09:21:20.000-04:00
diff --git 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,
