aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/ConstantLayer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/layers/ConstantLayer.cpp')
-rw-r--r--src/armnn/layers/ConstantLayer.cpp8
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