aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/armnn/SerializeLayerParameters.cpp6
-rw-r--r--src/armnn/SerializeLayerParameters.hpp5
2 files changed, 11 insertions, 0 deletions
diff --git a/src/armnn/SerializeLayerParameters.cpp b/src/armnn/SerializeLayerParameters.cpp
index b298e14ac3..0f110cc64a 100644
--- a/src/armnn/SerializeLayerParameters.cpp
+++ b/src/armnn/SerializeLayerParameters.cpp
@@ -11,6 +11,12 @@
namespace armnn
{
+void StringifyLayerParameters<ComparisonDescriptor>::Serialize(ParameterStringifyFunction& fn,
+ const ComparisonDescriptor& desc)
+{
+ fn("Operation", GetComparisonOperationAsCString(desc.m_Operation));
+}
+
void StringifyLayerParameters<PermuteDescriptor>::Serialize(ParameterStringifyFunction& fn,
const PermuteDescriptor& desc)
{
diff --git a/src/armnn/SerializeLayerParameters.hpp b/src/armnn/SerializeLayerParameters.hpp
index 20228e38ed..f760df0275 100644
--- a/src/armnn/SerializeLayerParameters.hpp
+++ b/src/armnn/SerializeLayerParameters.hpp
@@ -25,6 +25,11 @@ struct StringifyLayerParameters
static void Serialize(ParameterStringifyFunction&, const LayerParameter&) {}
};
+template <> struct StringifyLayerParameters<ComparisonDescriptor>
+{
+ static void Serialize(ParameterStringifyFunction& fn, const ComparisonDescriptor& desc);
+};
+
template <> struct StringifyLayerParameters<PermuteDescriptor>
{
static void Serialize(ParameterStringifyFunction& fn, const PermuteDescriptor& desc);