diff options
Diffstat (limited to 'src/armnnDeserializer/Deserializer.cpp')
-rw-r--r-- | src/armnnDeserializer/Deserializer.cpp | 2 |
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); } |