diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2020-11-11 18:01:48 +0000 |
|---|---|---|
| committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-11-12 11:35:11 +0000 |
| commit | 4189cc5ca4bb12e02c5e7f86ec6079f76d845b59 (patch) | |
| tree | bb5426804692e11abf000ffe9c64f7d95e104beb /delegate/src/FullyConnected.hpp | |
| parent | 8081536d24291794b4e189e6d5532d913a4525cb (diff) | |
| download | armnn-4189cc5ca4bb12e02c5e7f86ec6079f76d845b59.tar.gz | |
IVGCVSW-5504 'TfLiteDelegate: Introduce FP16 and BackendOptions'
* Added BackendOptions creations of armnn_delegate
* Included armnn/third-party the armnn_delegate unit tests
* Updated the CreateConstTensor function
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I8e2099a465766b905bff701413307e5850b68e42
Diffstat (limited to 'delegate/src/FullyConnected.hpp')
| -rw-r--r-- | delegate/src/FullyConnected.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/delegate/src/FullyConnected.hpp b/delegate/src/FullyConnected.hpp index f35f4c92b0..48bf06f94a 100644 --- a/delegate/src/FullyConnected.hpp +++ b/delegate/src/FullyConnected.hpp @@ -165,13 +165,13 @@ TfLiteStatus VisitFullyConnectedOperator(DelegateData& delegateData, biasTensorInfo, armnn::Optional<armnn::PermutationVector&>()); layer = delegateData.m_Network->AddFullyConnectedLayer(descriptor, - weightsTensor.first, - armnn::Optional<armnn::ConstTensor>(biasTensor.first)); + weightsTensor, + armnn::Optional<armnn::ConstTensor>(biasTensor)); } else { layer = delegateData.m_Network->AddFullyConnectedLayer(descriptor, - weightsTensor.first, + weightsTensor, armnn::EmptyOptional()); } ARMNN_ASSERT(layer != nullptr); |