Add THNN conversion of {ELU, LeakyReLU, LogSigmoid, LogSoftMax, LookupTable}

1 files changed, 13 insertions, 2 deletions

diff --git a/LogSigmoid.lua b/LogSigmoid.lua
index 7485ae6..b2b773a 100644
--- a/LogSigmoid.lua
+++ b/LogSigmoid.lua
@@ -6,9 +6,20 @@ function LogSigmoid:__init()
 end
 
 function LogSigmoid:updateOutput(input)
-   return input.nn.LogSigmoid_updateOutput(self, input)
+   input.THNN.LogSigmoid_updateOutput(
+      input:cdata(),
+      self.output:cdata(),
+      self.buffer:cdata()
+   )
+   return self.output
 end
 
 function LogSigmoid:updateGradInput(input, gradOutput)
-   return input.nn.LogSigmoid_updateGradInput(self, input, gradOutput)
+   input.THNN.LogSigmoid_updateGradInput(
+      input:cdata(),
+      gradOutput:cdata(),
+      self.gradInput:cdata(),
+      self.buffer:cdata()
+   )
+   return self.gradInput
 end