#include <vector>
#include "compute/operation.h"
#include "tensor/tensor.h"
#include "geadd_internal.h"
Go to the source code of this file.
|
template<typename T > |
AddOp< T > * | magmadnn::op::add (Operation< T > *a, Operation< T > *b, bool copy, bool needs_grad) |
|
- Author
- Daniel Nichols
- Version
- 1.0
- Date
- 2019-02-18
- Copyright
- Copyright (c) 2019
◆ add()
template<typename T >
AddOp< T > * magmadnn::op::add |
( |
Operation< T > * |
a, |
|
|
Operation< T > * |
b, |
|
|
bool |
copy = true , |
|
|
bool |
needs_grad = true |
|
) |
| |
Returns a new add operation (
- See also
- AddOp<T>).
- Template Parameters
-
- Parameters
-
a | |
b | |
copy | If copy is true then it returns a new tensor, if false then b=a+b. |
- Returns
- AddOp<T>*