sigmoid.hpp File Reference

Convolution implementations with NVidia cuDNN library. More...

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
template<typename I , typename C >
void	etl::impl::cudnn::sigmoid ([[maybe_unused]] I &&x, [[maybe_unused]] C &&y)
	Compute the sigmoid of x and store the result in y. More...
template<typename I , typename C >
void	etl::impl::cudnn::relu ([[maybe_unused]] I &&x, [[maybe_unused]] C &&y)
	Compute the RELU of x and store the result in y. More...
template<typename O , typename E , typename C >
void	etl::impl::cudnn::sigmoid_backward ([[maybe_unused]] O &&o, [[maybe_unused]] E &&e, [[maybe_unused]] C &&y)
	Compute the backward sigmoid of o/e and store the result in y. More...
template<typename O , typename E , typename C >
void	etl::impl::cudnn::relu_backward ([[maybe_unused]] O &&o, [[maybe_unused]] E &&e, [[maybe_unused]] C &&y)
	Compute the backward sigmoid of o/e and store the result in y. More...
template<typename I , typename C >
void	etl::impl::cudnn::softmax ([[maybe_unused]] I &&x, [[maybe_unused]] C &&y)
	Compute the softmax of x (batch) and store the result in y. More...
template<typename I , typename C >
void	etl::impl::cudnn::stable_softmax ([[maybe_unused]] I &&x, [[maybe_unused]] C &&y)
	Compute the softmax of x (batch) and store the result in y. More...

Detailed Description

Convolution implementations with NVidia cuDNN library.

Function Documentation

relu()

template<typename I , typename C >

void etl::impl::cudnn::relu	(	[[maybe_unused] ] I &&	x,
		[[maybe_unused] ] C &&	y
	)

Compute the RELU of x and store the result in y.

Parameters

x	The a expression
y	The c expression

relu_backward()

template<typename O , typename E , typename C >

void etl::impl::cudnn::relu_backward	(	[[maybe_unused] ] O &&	o,
		[[maybe_unused] ] E &&	e,
		[[maybe_unused] ] C &&	y
	)

Compute the backward sigmoid of o/e and store the result in y.

Parameters

o	The a expression
e	The b expression
y	The c expression

sigmoid()

template<typename I , typename C >

void etl::impl::cudnn::sigmoid	(	[[maybe_unused] ] I &&	x,
		[[maybe_unused] ] C &&	y
	)

Compute the sigmoid of x and store the result in y.

Parameters

x	The a expression
y	The c expression

sigmoid_backward()

template<typename O , typename E , typename C >

void etl::impl::cudnn::sigmoid_backward	(	[[maybe_unused] ] O &&	o,
		[[maybe_unused] ] E &&	e,
		[[maybe_unused] ] C &&	y
	)

Compute the backward sigmoid of o/e and store the result in y.

Parameters

o	The a expression
e	The b expression
y	The c expression

softmax()

template<typename I , typename C >

void etl::impl::cudnn::softmax	(	[[maybe_unused] ] I &&	x,
		[[maybe_unused] ] C &&	y
	)

Compute the softmax of x (batch) and store the result in y.

Parameters

x	The a expression
y	The c expression

stable_softmax()

template<typename I , typename C >

void etl::impl::cudnn::stable_softmax	(	[[maybe_unused] ] I &&	x,
		[[maybe_unused] ] C &&	y
	)

Compute the softmax of x (batch) and store the result in y.

Parameters

x	The a expression
y	The c expression