diff options
Diffstat (limited to 'utils/TypePrinter.h')
-rw-r--r-- | utils/TypePrinter.h | 37 |
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. |