diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-06-12 12:34:53 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2023-06-14 14:45:45 +0000 |
commit | 88644e62b8e621e4125f2d387ea82299b01dc40e (patch) | |
tree | abef6f850836fcc06d94869b5a324bf8056f0c76 | |
parent | 8eece0a80f03763b3fc0388adda776782f7147ff (diff) | |
download | armnn-88644e62b8e621e4125f2d387ea82299b01dc40e.tar.gz |
Add DataType to .dot files for constant layers
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I399ed1205f14d4fdd1194026c7a66bf00a1dd68d
-rw-r--r-- | src/armnn/layers/ConstantLayer.cpp | 8 | ||||
-rw-r--r-- | src/armnn/layers/ConstantLayer.hpp | 4 |
2 files changed, 10 insertions, 2 deletions
diff --git a/src/armnn/layers/ConstantLayer.cpp b/src/armnn/layers/ConstantLayer.cpp index aee95d063c..a17bb4fc93 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 @@ void ConstantLayer::ExecuteStrategy(IStrategy& strategy) const strategy.ExecuteStrategy(this, BaseDescriptor(), { layerOutputTensor }, GetName()); } +void ConstantLayer::SerializeLayerParameters(ParameterStringifyFunction& fn) const +{ + fn("DataType",GetDataTypeName(m_LayerOutput->GetTensorInfo().GetDataType())); + Layer::SerializeLayerParameters(fn); +} + } // namespace armnn diff --git a/src/armnn/layers/ConstantLayer.hpp b/src/armnn/layers/ConstantLayer.hpp index 08b9c24273..51938d8b53 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 @@ public: void ExecuteStrategy(IStrategy& strategy) const override; + void SerializeLayerParameters(ParameterStringifyFunction &fn) const override; + std::shared_ptr<ConstTensorHandle> m_LayerOutput; protected: |