Add functional conversion of HardTanh

1 files changed, 14 insertions, 0 deletions

diff --git a/THNN.lua b/THNN.lua
index 235b58c..dcae9d2 100644
--- a/THNN.lua
+++ b/THNN.lua
@@ -67,6 +67,20 @@ TH_API void THNN_(HardShrink_updateGradInput)(
           THTensor *gradOutput,
           THTensor *gradInput,
           real lambda);
+
+TH_API void THNN_(HardTanh_updateOutput)(
+          THNNState *state,
+          THTensor *input,
+          THTensor *output,
+          real min_val,
+          real max_val);
+TH_API void THNN_(HardTanh_updateGradInput)(
+          THNNState *state,
+          THTensor *input,
+          THTensor *gradOutput,
+          THTensor *gradInput,
+          real min_val,
+          real max_val);
 ]]
 
 -- THGenerator struct declaration copied from torch7/lib/TH/THRandom.h