diff options
Diffstat (limited to 'lib/THCUNN/generic/Sigmoid.cu')
-rw-r--r-- | lib/THCUNN/generic/Sigmoid.cu | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/lib/THCUNN/generic/Sigmoid.cu b/lib/THCUNN/generic/Sigmoid.cu index a4f7328..0e352d3 100644 --- a/lib/THCUNN/generic/Sigmoid.cu +++ b/lib/THCUNN/generic/Sigmoid.cu @@ -10,8 +10,7 @@ void THNN_(Sigmoid_updateOutput)( THCTensor *output) { THCUNN_assertSameGPU(state, 2, input, output); - THCTensor_(resizeAs)(state, output, input); - THC_pointwiseApply2(state, output, input, sigmoidupdateOutput_functor<real>()); + THCTensor_(sigmoid)(state, output, input); } void THNN_(Sigmoid_updateGradInput)( @@ -24,7 +23,7 @@ void THNN_(Sigmoid_updateGradInput)( THCUNN_check_nElement(state, input, gradOutput); THCUNN_assertSameGPU(state, 3, output, gradOutput, gradInput); THCTensor_(resizeAs)(state, gradInput, output); - THC_pointwiseApply3(state, gradInput, output, gradOutput, sigmoidupdateGradInput_functor<real>()); + THC_pointwiseApply3(state, gradInput, output, gradOutput, SigmoidGradInputOp<real>()); } #endif |