Expression Templates Library (ETL)
Functions
bias_add.hpp File Reference

EGBLAS wrappers for the bias_add_2d operation. More...

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

Go to the source code of this file.

Functions

void etl::impl::egblas::bias_add_2d ([[maybe_unused]] size_t m, [[maybe_unused]] size_t n, [[maybe_unused]] const float *x, [[maybe_unused]] size_t incx, [[maybe_unused]] const float *b, [[maybe_unused]] size_t incb, [[maybe_unused]] float *y, [[maybe_unused]] size_t incy)
 Add the 1D bias to to the batched 1D input. More...
 
void etl::impl::egblas::bias_add_2d ([[maybe_unused]] size_t m, [[maybe_unused]] size_t n, [[maybe_unused]] const double *x, [[maybe_unused]] size_t incx, [[maybe_unused]] const double *b, [[maybe_unused]] size_t incb, [[maybe_unused]] double *y, [[maybe_unused]] size_t incy)
 Add the 1D bias to to the batched 1D input. More...
 
void etl::impl::egblas::bias_add_4d ([[maybe_unused]] size_t m, [[maybe_unused]] size_t n, [[maybe_unused]] size_t o, [[maybe_unused]] size_t p, [[maybe_unused]] const float *x, [[maybe_unused]] size_t incx, [[maybe_unused]] const float *b, [[maybe_unused]] size_t incb, [[maybe_unused]] float *y, [[maybe_unused]] size_t incy)
 Add the 1D bias to to the batched 1D input. More...
 
void etl::impl::egblas::bias_add_4d ([[maybe_unused]] size_t m, [[maybe_unused]] size_t n, [[maybe_unused]] size_t o, [[maybe_unused]] size_t p, [[maybe_unused]] const double *x, [[maybe_unused]] size_t incx, [[maybe_unused]] const double *b, [[maybe_unused]] size_t incb, [[maybe_unused]] double *y, [[maybe_unused]] size_t incy)
 Add the 1D bias to to the batched 1D input. More...
 

Detailed Description

EGBLAS wrappers for the bias_add_2d operation.

Function Documentation

◆ bias_add_2d() [1/2]

void etl::impl::egblas::bias_add_2d ( [[maybe_unused] ] size_t  m,
[[maybe_unused] ] size_t  n,
[[maybe_unused] ] const float *  x,
[[maybe_unused] ] size_t  incx,
[[maybe_unused] ] const float *  b,
[[maybe_unused] ] size_t  incb,
[[maybe_unused] ] float *  y,
[[maybe_unused] ] size_t  incy 
)
inline

Add the 1D bias to to the batched 1D input.

Parameters
mThe first dimension of x and y
nThe second dimension of x and y
xThe vector x (GPU memory)
incxThe stride of x
bThe vector x (GPU memory)
incbThe stride of x
yThe vector y (GPU memory)
incyThe stride of y

◆ bias_add_2d() [2/2]

void etl::impl::egblas::bias_add_2d ( [[maybe_unused] ] size_t  m,
[[maybe_unused] ] size_t  n,
[[maybe_unused] ] const double *  x,
[[maybe_unused] ] size_t  incx,
[[maybe_unused] ] const double *  b,
[[maybe_unused] ] size_t  incb,
[[maybe_unused] ] double *  y,
[[maybe_unused] ] size_t  incy 
)
inline

Add the 1D bias to to the batched 1D input.

Parameters
mThe first dimension of x and y
nThe second dimension of x and y
xThe vector x (GPU memory)
incxThe stride of x
bThe vector x (GPU memory)
incbThe stride of x
yThe vector y (GPU memory)
incyThe stride of y

◆ bias_add_4d() [1/2]

void etl::impl::egblas::bias_add_4d ( [[maybe_unused] ] size_t  m,
[[maybe_unused] ] size_t  n,
[[maybe_unused] ] size_t  o,
[[maybe_unused] ] size_t  p,
[[maybe_unused] ] const float *  x,
[[maybe_unused] ] size_t  incx,
[[maybe_unused] ] const float *  b,
[[maybe_unused] ] size_t  incb,
[[maybe_unused] ] float *  y,
[[maybe_unused] ] size_t  incy 
)
inline

Add the 1D bias to to the batched 1D input.

Parameters
mThe first dimension of x and y
nThe second dimension of x and y
xThe vector x (GPU memory)
incxThe stride of x
bThe vector x (GPU memory)
incbThe stride of x
yThe vector y (GPU memory)
incyThe stride of y

◆ bias_add_4d() [2/2]

void etl::impl::egblas::bias_add_4d ( [[maybe_unused] ] size_t  m,
[[maybe_unused] ] size_t  n,
[[maybe_unused] ] size_t  o,
[[maybe_unused] ] size_t  p,
[[maybe_unused] ] const double *  x,
[[maybe_unused] ] size_t  incx,
[[maybe_unused] ] const double *  b,
[[maybe_unused] ] size_t  incb,
[[maybe_unused] ] double *  y,
[[maybe_unused] ] size_t  incy 
)
inline

Add the 1D bias to to the batched 1D input.

Parameters
mThe first dimension of x and y
nThe second dimension of x and y
xThe vector x (GPU memory)
incxThe stride of x
bThe vector x (GPU memory)
incbThe stride of x
yThe vector y (GPU memory)
incyThe stride of y