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

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