#include <vector>
#include "layer/layer.h"
#include "tensor/tensor.h"
#include "compute/operation.h"
#include "compute/tensor_operations.h"
Go to the source code of this file.
Classes | |
class | magmadnn::layer::ActivationLayer< T > |
Enumerations | |
enum | activation_t { SIGMOID, TANH, RELU, SOFTMAX } |
Functions | |
template<typename T > | |
ActivationLayer< T > * | magmadnn::layer::activation (op::Operation< T > *input, activation_t activation_func) |