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