MagmaDNN  1.0
c++NeuralNetworkFramework
geadd_internal.h File Reference
#include "cblas.h"
#include "tensor/tensor.h"
Include dependency graph for geadd_internal.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions

template<typename T >
bool magmadnn::internal::geadd_check (Tensor< T > *A, Tensor< T > *B, Tensor< T > *C)
 
template<typename T >
void magmadnn::internal::geadd_full (T alpha, Tensor< T > *A, T beta, Tensor< T > *B, Tensor< T > *C)
 
template<typename T >
void magmadnn::internal::tensor_scalar_add_full (T alpha, Tensor< T > *x, Tensor< T > *out)
 

Detailed Description

Author
Daniel Nichols
Version
1.0
Date
2019-02-22

Function Documentation

◆ 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
T
Parameters
Aa tensor
Ba tensor
Ca tensor
Returns
true
false
Here is the call graph for this function:
Here is the caller graph for this function:

◆ geadd_full()

template<typename T >
void magmadnn::internal::geadd_full ( alpha,
Tensor< T > *  A,
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
Tint, float, or double
Parameters
alphascaling value
Aa tensor
betascaling value
Ba tensor
Ca tensor

◆ tensor_scalar_add_full()

template<typename T >
void magmadnn::internal::tensor_scalar_add_full ( alpha,
Tensor< T > *  x,
Tensor< T > *  out 
)
Template Parameters
Tnumeric
Parameters
alpha
x
out