1 files changed, 22 insertions, 0 deletions
diff --git a/tests/TypePrinter.h b/tests/TypePrinter.h
index 5f7313552e..020b559c6f 100644
--- a/tests/TypePrinter.h
+++ b/tests/TypePrinter.h
@@ -198,6 +198,21 @@ inline ::std::ostream &operator<<(::std::ostream &os, const ConvertPolicy &polic
     return os;
 }
 
+/** Formatted output of the Reduction Operations. */
+inline ::std::ostream &operator<<(::std::ostream &os, const ReductionOperation &op)
+{
+    switch(op)
+    {
+        case ReductionOperation::SUM_SQUARE:
+            os << "SUM_SQUARE";
+            break;
+        default:
+            ARM_COMPUTE_ERROR("NOT_SUPPORTED!");
+    }
+
+    return os;
+}
+
 /** Formatted output of the activation function type. */
 inline ::std::ostream &operator<<(::std::ostream &os, const ActivationLayerInfo::ActivationFunction &act_function)
 {
@@ -240,6 +255,13 @@ inline ::std::ostream &operator<<(::std::ostream &os, const ActivationLayerInfo:
     return os;
 }
 
+inline std::string to_string(const ReductionOperation &op)
+{
+    std::stringstream str;
+    str << op;
+    return str.str();
+}
+
 inline std::string to_string(const ActivationLayerInfo::ActivationFunction &function)
 {
     std::stringstream str;