#include "FullyConnected.hpp"
#include "RefWorkloadUtils.hpp"
#include <boost/assert.hpp>
Go to the source code of this file.
|
| armnn |
| Copyright (c) 2020 ARM Limited.
|
|
|
void | FullyConnected (const TensorShape &rInputShape, Decoder< float > &rInputDecoder, const TensorShape &rOutputShape, Encoder< float > &rOutputEncoder, Decoder< float > &rWeightDecoder, Decoder< float > &rBiasDecoder, bool biasEnabled, unsigned int K, bool transposeWeights) |
| Performs a matrix multiplication and optionally adds a bias. More...
|
|