#include "cblas.h"
#include "tensor/tensor.h"
Go to the source code of this file.
- Author
- Daniel Nichols
- Version
- 1.0
- Date
- 2019-02-22
- Copyright
- Copyright (c) 2019
◆ geadd_check()
template<typename T >
bool magmadnn::internal::geadd_check |
( |
Tensor< T > * |
A, |
|
|
Tensor< T > * |
B, |
|
|
Tensor< T > * |
C |
|
) |
| |
Returns true if A, B, C are valid parameters to geadd_full.
- Template Parameters
-
- Parameters
-
A | a tensor |
B | a tensor |
C | a tensor |
- Returns
- true
-
false
◆ geadd_full()
template<typename T >
void magmadnn::internal::geadd_full |
( |
T |
alpha, |
|
|
Tensor< T > * |
A, |
|
|
T |
beta, |
|
|
Tensor< T > * |
B, |
|
|
Tensor< T > * |
C |
|
) |
| |
Computes C = alpha*A + beta*B All tensors must have the same memory type and shape/size.
- Template Parameters
-
- Parameters
-
alpha | scaling value |
A | a tensor |
beta | scaling value |
B | a tensor |
C | a tensor |
◆ tensor_scalar_add_full()
template<typename T >
void magmadnn::internal::tensor_scalar_add_full |
( |
T |
alpha, |
|
|
Tensor< T > * |
x, |
|
|
Tensor< T > * |
out |
|
) |
| |
- Template Parameters
-
- Parameters
-