aboutsummaryrefslogtreecommitdiff
path: root/utils/TypePrinter.h
diff options
context:
space:
mode:
Diffstat (limited to 'utils/TypePrinter.h')
-rw-r--r--utils/TypePrinter.h37
1 files changed, 37 insertions, 0 deletions
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h
index 27560e6b07..6850ea018a 100644
--- a/utils/TypePrinter.h
+++ b/utils/TypePrinter.h
@@ -1453,6 +1453,30 @@ inline ::std::ostream &operator<<(::std::ostream &os, const ComparisonOperation
return os;
}
+/** Formatted output of the Elementwise unary Operations.
+ *
+ * @param[out] os Output stream.
+ * @param[in] op Type to output.
+ *
+ * @return Modified output stream.
+ */
+inline ::std::ostream &operator<<(::std::ostream &os, const ElementWiseUnary &op)
+{
+ switch(op)
+ {
+ case ElementWiseUnary::RSQRT:
+ os << "RSQRT";
+ break;
+ case ElementWiseUnary::EXP:
+ os << "EXP";
+ break;
+ default:
+ ARM_COMPUTE_ERROR("NOT_SUPPORTED!");
+ }
+
+ return os;
+}
+
/** Formatted output of the Comparison Operations.
*
* @param[in] op Type to output.
@@ -1466,6 +1490,19 @@ inline std::string to_string(const ComparisonOperation &op)
return str.str();
}
+/** Formatted output of the Elementwise unary Operations.
+ *
+ * @param[in] op Type to output.
+ *
+ * @return Formatted string.
+ */
+inline std::string to_string(const ElementWiseUnary &op)
+{
+ std::stringstream str;
+ str << op;
+ return str.str();
+}
+
/** Formatted output of the Norm Type.
*
* @param[in] type Type to output.