EGBLAS wrappers for the bias_add_2d operation.
More...
Go to the source code of this file.
|
| 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...
|
| |
EGBLAS wrappers for the bias_add_2d operation.
◆ 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
-
| m | The first dimension of x and y |
| n | The second dimension of x and y |
| x | The vector x (GPU memory) |
| incx | The stride of x |
| b | The vector x (GPU memory) |
| incb | The stride of x |
| y | The vector y (GPU memory) |
| incy | The 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
-
| m | The first dimension of x and y |
| n | The second dimension of x and y |
| x | The vector x (GPU memory) |
| incx | The stride of x |
| b | The vector x (GPU memory) |
| incb | The stride of x |
| y | The vector y (GPU memory) |
| incy | The 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
-
| m | The first dimension of x and y |
| n | The second dimension of x and y |
| x | The vector x (GPU memory) |
| incx | The stride of x |
| b | The vector x (GPU memory) |
| incb | The stride of x |
| y | The vector y (GPU memory) |
| incy | The 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
-
| m | The first dimension of x and y |
| n | The second dimension of x and y |
| x | The vector x (GPU memory) |
| incx | The stride of x |
| b | The vector x (GPU memory) |
| incb | The stride of x |
| y | The vector y (GPU memory) |
| incy | The stride of y |