Add gradWeightBuf & gradWeightBuf2 params to PReLU_accGradParameters

1 files changed, 6 insertions, 4 deletions

diff --git a/THNN.lua b/THNN.lua
index 11ded8f..a72e961 100644
--- a/THNN.lua
+++ b/THNN.lua
@@ -159,15 +159,15 @@ TH_API void THNN_(MarginCriterion_updateOutput)(
           THTensor *input,
           THTensor *target,
           THTensor *output,
-          real margin,
-          bool sizeAverage);
+          bool sizeAverage,
+          real margin);
 TH_API void THNN_(MarginCriterion_updateGradInput)(
           THNNState *state,
           THTensor *input,
           THTensor *target,
           THTensor *gradInput,
-          real margin,
-          bool sizeAverage);
+          bool sizeAverage,
+          real margin);
 
 TH_API void THNN_(MSECriterion_updateOutput)(
           THNNState *state,
@@ -230,6 +230,8 @@ TH_API void THNN_(PReLU_accGradParameters)(
           THTensor* gradInput,
           THTensor *weight,
           THTensor *gradWeight,
+          THTensor *gradWeightBuf,
+          THTensor *gradWeightBuf2,
           THIndex_t nOutputPlane,
           real scale);