#include "compute/operation.h"
#include "math/relu.h"
#include "compute/relu/relu_internal.h"
#include "tensor/tensor.h"
Go to the source code of this file.
Classes | |
class | magmadnn::op::ReluOp< T > |
Functions | |
template<typename T > | |
ReluOp< T > * | magmadnn::op::relu (Operation< T > *x, bool copy, bool needs_grad) |