aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/SerializeLayerParameters.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/SerializeLayerParameters.cpp')
-rw-r--r--src/armnn/SerializeLayerParameters.cpp20
1 files changed, 20 insertions, 0 deletions
diff --git a/src/armnn/SerializeLayerParameters.cpp b/src/armnn/SerializeLayerParameters.cpp
index e4bf094b7c..b298e14ac3 100644
--- a/src/armnn/SerializeLayerParameters.cpp
+++ b/src/armnn/SerializeLayerParameters.cpp
@@ -135,6 +135,26 @@ void StringifyLayerParameters<Pooling2dDescriptor>::Serialize(ParameterStringify
fn("DataLayout", GetDataLayoutName(desc.m_DataLayout));
}
+void StringifyLayerParameters<ReduceDescriptor>::Serialize(ParameterStringifyFunction& fn, const ReduceDescriptor& desc)
+{
+ {
+ std::stringstream ss;
+ int count = 0;
+ for (auto&& var : desc.m_vAxis)
+ {
+ if (count > 0)
+ {
+ ss << ",";
+ }
+ ss << var;
+ ++count;
+ }
+ fn("Axis", ss.str());
+ }
+ fn("KeepDims", (desc.m_KeepDims ? "true" : "false"));
+ fn("ReduceOperation", GetReduceOperationAsCString(desc.m_ReduceOperation));
+}
+
void StringifyLayerParameters<SoftmaxDescriptor>::Serialize(ParameterStringifyFunction& fn,
const SoftmaxDescriptor& desc)
{