fix leaky relu header_...

710c4ff0 · Benjamin Thomas Graham · 209e1e78 · 710c4ff0
Commit 710c4ff0 authored Oct 03, 2017 by Benjamin Thomas Graham
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

Torch/C.lua Torch/C.lua +2 -2

No files found.
--- a/Torch/C.lua
+++ b/Torch/C.lua
@@ -108,11 +108,11 @@ void scn_ARCH_REAL_BatchNormalizationInTensor_updateOutput(

 // LeakyReLU
 void scn_ARCH_REAL_LeakyReLU_updateOutput(
-  THTensor *input_features, THTensor *output_features, long n,
+  THTensor *input_features, THTensor *output_features,
  float alpha);
 void scn_ARCH_REAL_LeakyReLU_updateGradInput(
  THTensor *input_features, THTensor *d_input_features,
-  THTensor *d_output_features, long n, float alpha);
+  THTensor *d_output_features, float alpha);

 // NetworkInNetwork
 double scn_ARCH_REAL_NetworkInNetwork_updateOutput(