sigmoid_internal.cpp File Reference

#include "compute/sigmoid/sigmoid_internal.h"

Include dependency graph for sigmoid_internal.cpp:

Functions
template<typename T >
void	magmadnn::internal::sigmoid_full (Tensor< T > *x, Tensor< T > *out, bool fast)
template void	magmadnn::internal::sigmoid_full (Tensor< int > *x, Tensor< int > *out, bool fast)
template void	magmadnn::internal::sigmoid_full (Tensor< float > *x, Tensor< float > *out, bool fast)
template void	magmadnn::internal::sigmoid_full (Tensor< double > *x, Tensor< double > *out, bool fast)
template<typename T >
void	magmadnn::internal::sigmoid_grad (Tensor< T > *output, Tensor< T > *grad, Tensor< T > *out)
template void	magmadnn::internal::sigmoid_grad (Tensor< int > *output, Tensor< int > *grad, Tensor< int > *out)
template void	magmadnn::internal::sigmoid_grad (Tensor< float > *output, Tensor< float > *grad, Tensor< float > *out)
template void	magmadnn::internal::sigmoid_grad (Tensor< double > *output, Tensor< double > *grad, Tensor< double > *out)

Detailed Description

Author

Daniel Nichols

Version

0.1

Date

2019-02-23

Copyright

Copyright (c) 2019

Function Documentation

sigmoid_full()

template<typename T >

void magmadnn::internal::sigmoid_full	(	Tensor< T > *	x,
		Tensor< T > *	out,
		bool	fast = true
	)

Computes the element-wise sigmoid on x.

Template Parameters

T

Parameters

x	pointer to tensor to be sigmoided
fast	if true, then x=1/(1+|x|) is computed instead of normal sigmoid