diff options
author | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-14 10:36:13 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-14 13:33:59 +0100 |
commit | fc598e15ff30bc375c95c9536d4a56662d867926 (patch) | |
tree | 0d17a7928ae4faab6978552e666123bfc1926d93 /src/armnnDeserializer/Deserializer.cpp | |
parent | 906f94631aa7ef590b9d8ff45507e818a0d1ac2c (diff) | |
download | armnn-fc598e15ff30bc375c95c9536d4a56662d867926.tar.gz |
Use the new deprecation API
* Used the new ARMNN_DEPRECATED_MSG macro instead of @deprecated
* Refactored the code to no longer use the deprecated methods where
applicable
!android-nn-driver:1126
Change-Id: Ib0578d3d6fc5a763f5fb922f67ba91fafc7796f6
Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
Diffstat (limited to 'src/armnnDeserializer/Deserializer.cpp')
-rw-r--r-- | src/armnnDeserializer/Deserializer.cpp | 24 |
1 files changed, 12 insertions, 12 deletions
diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp index b7d45e0a7d..170917e5cf 100644 --- a/src/armnnDeserializer/Deserializer.cpp +++ b/src/armnnDeserializer/Deserializer.cpp @@ -988,13 +988,15 @@ void Deserializer::ParseConvolution2d(GraphPtr graph, unsigned int layerIndex) armnn::ConstTensor weights = ToConstTensor(serializerLayer->weights()); armnn::ConstTensor biases; + armnn::Optional<armnn::ConstTensor> optionalBiases = armnn::EmptyOptional(); if (descriptor.m_BiasEnabled) { biases = ToConstTensor(serializerLayer->biases()); + optionalBiases = armnn::Optional<armnn::ConstTensor>(biases); } IConnectableLayer* layer = m_Network->AddConvolution2dLayer(descriptor, weights, - biases, + optionalBiases, layerName.c_str()); armnn::TensorInfo outputTensorInfo = ToTensorInfo(outputs[0]); layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); @@ -1030,13 +1032,15 @@ void Deserializer::ParseDepthwiseConvolution2d(GraphPtr graph, unsigned int laye armnn::ConstTensor weights = ToConstTensor(serializerLayer->weights()); armnn::ConstTensor biases; + armnn::Optional<armnn::ConstTensor> optionalBiases = armnn::EmptyOptional(); if (descriptor.m_BiasEnabled) { biases = ToConstTensor(serializerLayer->biases()); + optionalBiases = armnn::Optional<armnn::ConstTensor>(biases); } IConnectableLayer* layer = m_Network->AddDepthwiseConvolution2dLayer(descriptor, weights, - biases, + optionalBiases, layerName.c_str()); armnn::TensorInfo outputTensorInfo = ToTensorInfo(outputs[0]); @@ -1317,20 +1321,16 @@ void Deserializer::ParseFullyConnected(GraphPtr graph, unsigned int layerIndex) armnn::ConstTensor weightsTensor = ToConstTensor(flatBufferLayer->weights()); armnn::IConnectableLayer* layer; + armnn::Optional<armnn::ConstTensor> optionalBiases = armnn::EmptyOptional(); if (flatBufferDescriptor->biasEnabled()) { armnn::ConstTensor biasTensorData = ToConstTensor(flatBufferLayer->biases()); - layer = m_Network->AddFullyConnectedLayer(fullyConnectedDescriptor, - weightsTensor, - biasTensorData, - layerName.c_str()); - } - else - { - layer = m_Network->AddFullyConnectedLayer(fullyConnectedDescriptor, - weightsTensor, - layerName.c_str()); + optionalBiases = armnn::Optional<armnn::ConstTensor>(biasTensorData); } + layer = m_Network->AddFullyConnectedLayer(fullyConnectedDescriptor, + weightsTensor, + optionalBiases, + layerName.c_str()); armnn::TensorInfo outputTensorInfo = ToTensorInfo(outputs[0]); layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); |