#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::OutputLayer< T > |
Functions | |
template<typename T > | |
OutputLayer< T > * | magmadnn::layer::output (op::Operation< T > *input) |