MagmaDNN  1.0
c++NeuralNetworkFramework
sigmoid_internal.h
Go to the documentation of this file.
1 
9 #pragma once
10 #include <math.h>
11 #include "tensor/tensor.h"
12 
13 
14 namespace magmadnn {
15 namespace internal {
16 
22 template <typename T>
23 void sigmoid_full(Tensor<T> *x, Tensor<T> *out, bool fast=true);
24 
25 #if defined(_HAS_CUDA_)
26 
31 template <typename T>
32 void sigmoid_full_device(Tensor<T> *x, Tensor<T> *out, bool fast=true);
33 #endif
34 
35 
36 
37 template <typename T>
38 void sigmoid_grad(Tensor<T> *output, Tensor<T> *grad, Tensor<T> *out);
39 
40 #if defined(_HAS_CUDA_)
41 template <typename T>
42 void sigmoid_grad_device(Tensor<T> *output, Tensor<T> *grad, Tensor<T> *out);
43 #endif
44 
45 
46 } // namespace internal
47 } // namespace magmadnn
magmadnn
Definition: addop.cpp:11
magmadnn::internal::sigmoid_full
void sigmoid_full(Tensor< T > *x, Tensor< T > *out, bool fast)
Definition: sigmoid_internal.cpp:15