ArmNN
 24.02
FullyConnected.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include "BaseIterator.hpp"
9 #include "Decoders.hpp"
10 #include "Encoders.hpp"
11 #include <armnn/Tensor.hpp>
13 
14 namespace armnn
15 {
16 
17 /// Performs a matrix multiplication and optionally adds a bias.
18 void FullyConnected(const TensorShape& rInputShape,
19  Decoder<float>& rInputDecoder,
20  const TensorShape& rOutputShape,
21  Encoder<float>& rOutputEncoder,
22  const TensorShape& rWeightsShape,
23  Decoder<float>& rWeightDecoder,
24  Decoder<float>* rBiasDecoder,
25  bool biasEnabled,
26  unsigned int K,
27  bool transposeWeights);
28 
29 } //namespace armnn
WorkloadData.hpp
BaseIterator.hpp
armnn::FullyConnected
void FullyConnected(const TensorShape &rInputShape, Decoder< float > &rInputDecoder, const TensorShape &rOutputShape, Encoder< float > &rOutputEncoder, const TensorShape &rWeightsShape, Decoder< float > &rWeightDecoder, Decoder< float > *pBiasDecoder, const bool biasEnabled, const unsigned int K, const bool transposeWeights)
Performs a matrix multiplication and optionally adds a bias.
Definition: FullyConnected.cpp:15
Tensor.hpp
Decoders.hpp
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
Encoders.hpp