From 361679da2a86b6794827edf48a3f585738f8e08a Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Tue, 2 Aug 2022 09:17:23 +0100 Subject: IVGCVSW-7063 'Support Library NNAPI Caching' * Fixed caching issue. Signed-off-by: Sadik Armagan Change-Id: Ic7b3e0bd4438b2fd1b3dbfa86b6c89d625bbf9dd --- src/armnnDeserializer/Deserializer.cpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/armnnDeserializer/Deserializer.cpp') 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); } -- cgit v1.2.1