From 88644e62b8e621e4125f2d387ea82299b01dc40e Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Mon, 12 Jun 2023 12:34:53 +0100 Subject: Add DataType to .dot files for constant layers Signed-off-by: Teresa Charlin Change-Id: I399ed1205f14d4fdd1194026c7a66bf00a1dd68d --- src/armnn/layers/ConstantLayer.cpp | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'src/armnn/layers/ConstantLayer.cpp') 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 -- cgit v1.2.1