diff options
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); |