commit 88644e62b8e621e4125f2d387ea82299b01dc40e
author Teresa Charlin <teresa.charlinreyes@arm.com> Mon Jun 12 12:34:53 2023 +0100
committer TeresaARM <teresa.charlinreyes@arm.com> Wed Jun 14 14:45:45 2023 +0000
tree abef6f850836fcc06d94869b5a324bf8056f0c76
parent 8eece0a80f03763b3fc0388adda776782f7147ff

Add DataType to .dot files for constant layers


Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I399ed1205f14d4fdd1194026c7a66bf00a1dd68d

src/armnn/layers/ConstantLayer.cpp

diff --git a/src/armnn/layers/ConstantLayer.cpp b/src/armnn/layers/ConstantLayer.cpp
index aee95d0..a17bb4f 100644
--- a/src/armnn/layers/ConstantLayer.cpp
+++ b/src/armnn/layers/ConstantLayer.cpp
@@ -1,5 +1,5 @@
 //
-// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
+// Copyright © 2017, 2023 Arm Ltd and Contributors. All rights reserved.
 // SPDX-License-Identifier: MIT
 //
 #include "ConstantLayer.hpp"
@@ -69,4 +69,10 @@
     strategy.ExecuteStrategy(this, BaseDescriptor(), { layerOutputTensor }, GetName());
 }
 
+void ConstantLayer::SerializeLayerParameters(ParameterStringifyFunction& fn) const
+{
+    fn("DataType",GetDataTypeName(m_LayerOutput->GetTensorInfo().GetDataType()));
+    Layer::SerializeLayerParameters(fn);
+}
+
 } // namespace armnn

src/armnn/layers/ConstantLayer.hpp

diff --git a/src/armnn/layers/ConstantLayer.hpp b/src/armnn/layers/ConstantLayer.hpp
index 08b9c24..51938d8 100644
--- a/src/armnn/layers/ConstantLayer.hpp
+++ b/src/armnn/layers/ConstantLayer.hpp
@@ -1,5 +1,5 @@
 //
-// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
+// Copyright © 2017, 2023 Arm Ltd and Contributors. All rights reserved.
 // SPDX-License-Identifier: MIT
 //
 #pragma once
@@ -41,6 +41,8 @@
 
     void ExecuteStrategy(IStrategy& strategy) const override;
 
+    void SerializeLayerParameters(ParameterStringifyFunction &fn) const override;
+
     std::shared_ptr<ConstTensorHandle> m_LayerOutput;
 
 protected: