geadd_internal.cpp File Reference

#include "compute/add/geadd_internal.h"

Include dependency graph for geadd_internal.cpp:

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 void	magmadnn::internal::geadd_full (int alpha, Tensor< int > *A, int beta, Tensor< int > *B, Tensor< int > *C)
template void	magmadnn::internal::geadd_full (float alpha, Tensor< float > *A, float beta, Tensor< float > *B, Tensor< float > *C)
template void	magmadnn::internal::geadd_full (double alpha, Tensor< double > *A, double beta, Tensor< double > *B, Tensor< double > *C)
template<typename T >
void	magmadnn::internal::tensor_scalar_add_full (T alpha, Tensor< T > *x, Tensor< T > *out)
template void	magmadnn::internal::tensor_scalar_add_full (int alpha, Tensor< int > *x, Tensor< int > *out)
template void	magmadnn::internal::tensor_scalar_add_full (float alpha, Tensor< float > *x, Tensor< float > *out)
template void	magmadnn::internal::tensor_scalar_add_full (double alpha, Tensor< double > *x, Tensor< double > *out)

Detailed Description

Author

Daniel Nichols

Version

1.0

Date

2019-02-22

Copyright

Copyright (c) 2019

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

A	a tensor
B	a tensor
C	a 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	(	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

T	int, float, or double

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

T

numeric

Parameters

alpha
x
out