bias_batch_mean.hpp File Reference

bias_batch_mean 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 X , typename Y >
void	etl::impl::cudnn::bias_batch_mean_4d ([[maybe_unused]] X &&x, [[maybe_unused]] Y &&y)
	CUDNN Implementation of the bias_batch_mean operation. More...
template<typename X , typename Y >
void	etl::impl::cudnn::bias_batch_mean_2d ([[maybe_unused]] X &&x, [[maybe_unused]] Y &&y)
	CUDNN Implementation of the bias_batch_mean operation. More...

Detailed Description

bias_batch_mean implementations with NVidia cuDNN library

Function Documentation

bias_batch_mean_2d()

template<typename X , typename Y >

void etl::impl::cudnn::bias_batch_mean_2d	(	[[maybe_unused] ] X &&	x,
		[[maybe_unused] ] Y &&	y
	)

CUDNN Implementation of the bias_batch_mean operation.

Parameters

x	The input tensor
y	The output tensor

bias_batch_mean_4d()

template<typename X , typename Y >

void etl::impl::cudnn::bias_batch_mean_4d	(	[[maybe_unused] ] X &&	x,
		[[maybe_unused] ] Y &&	y
	)

CUDNN Implementation of the bias_batch_mean operation.

Parameters

x	The input tensor
y	The output tensor