aboutsummaryrefslogtreecommitdiff
path: root/src/armnnDeserializer/Deserializer.cpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2022-08-02 09:17:23 +0100
committerNikhil Raj <nikhil.raj@arm.com>2022-08-05 15:51:44 +0100
commit361679da2a86b6794827edf48a3f585738f8e08a (patch)
tree39fe40e4918ccdf96241307a228d8010ba62cd8a /src/armnnDeserializer/Deserializer.cpp
parent8950f8a59dbfea302dfaa2a5df320bee5d877745 (diff)
downloadarmnn-361679da2a86b6794827edf48a3f585738f8e08a.tar.gz
IVGCVSW-7063 'Support Library NNAPI Caching'
* Fixed caching issue. Signed-off-by: Sadik Armagan <sadik.armagan@arm.com> Change-Id: Ic7b3e0bd4438b2fd1b3dbfa86b6c89d625bbf9dd
Diffstat (limited to 'src/armnnDeserializer/Deserializer.cpp')
-rw-r--r--src/armnnDeserializer/Deserializer.cpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp
index 90558bbd53..a405cb92a5 100644
--- a/src/armnnDeserializer/Deserializer.cpp
+++ b/src/armnnDeserializer/Deserializer.cpp
@@ -1396,6 +1396,7 @@ void IDeserializer::DeserializerImpl::ParseConstant(GraphPtr graph, unsigned int
weightsShape[0],
weightsShape[1],
weightsShape[2]*weightsShape[3]});
+ weightsInfo.SetConstant(true);
armnn::ConstTensor weightsPermuted(weightsInfo, permuteBuffer.get());
@@ -1412,6 +1413,7 @@ void IDeserializer::DeserializerImpl::ParseConstant(GraphPtr graph, unsigned int
layer = m_Network->AddConstantLayer(input, layerName.c_str());
armnn::TensorInfo outputTensorInfo = ToTensorInfo(outputs[0]);
+ outputTensorInfo.SetConstant(true);
layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo);
}