bias_add.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 K , typename C >
void	etl::impl::cudnn::bias_add_4d ([[maybe_unused]] I &&x, [[maybe_unused]] K &&b, [[maybe_unused]] C &&y)
	Compute the bias addition of b into x and store the result in y. More...
template<typename I , typename K , typename C >
void	etl::impl::cudnn::bias_add_2d ([[maybe_unused]] I &&x, [[maybe_unused]] K &&b, [[maybe_unused]] C &&y)
	Compute the bias addition of b into x and store the result in y. More...

Detailed Description

Convolution implementations with NVidia cuDNN library.

Function Documentation

bias_add_2d()

template<typename I , typename K , typename C >

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

Compute the bias addition of b into x and store the result in y.

Parameters

x	The a expression
b	The b expression
y	The c expression

bias_add_4d()

template<typename I , typename K , typename C >

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

Compute the bias addition of b into x and store the result in y.

Parameters

x	The a expression
b	The b expression
y	The c expression