aboutsummaryrefslogtreecommitdiff
path: root/tests/TypePrinter.h
diff options
context:
space:
mode:
Diffstat (limited to 'tests/TypePrinter.h')
-rw-r--r--tests/TypePrinter.h7
1 files changed, 7 insertions, 0 deletions
diff --git a/tests/TypePrinter.h b/tests/TypePrinter.h
index 10d33882ce..5f7313552e 100644
--- a/tests/TypePrinter.h
+++ b/tests/TypePrinter.h
@@ -240,6 +240,13 @@ inline ::std::ostream &operator<<(::std::ostream &os, const ActivationLayerInfo:
return os;
}
+inline std::string to_string(const ActivationLayerInfo::ActivationFunction &function)
+{
+ std::stringstream str;
+ str << function;
+ return str.str();
+}
+
inline std::string to_string(const ActivationLayerInfo &info)
{
std::stringstream str;