#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::FullyConnectedLayer< T > |
Functions | |
template<typename T > | |
FullyConnectedLayer< T > * | magmadnn::layer::fullyconnected (op::Operation< T > *input, unsigned int hidden_units, bool use_bias) |