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 /src/armnn/layers/ConstantLayer.cpp | |
| 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
Diffstat (limited to 'src/armnn/layers/ConstantLayer.cpp')
| -rw-r--r-- | src/armnn/layers/ConstantLayer.cpp | 8 |
1 files changed, 7 insertions, 1 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 |