diff options
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/TypePrinter.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h index b072d1a308..2fa3d0ea4b 100644 --- a/utils/TypePrinter.h +++ b/utils/TypePrinter.h @@ -311,6 +311,13 @@ inline ::std::ostream &operator<<(::std::ostream &os, const PoolingLayerInfo &in return os; } +inline std::string to_string(const RoundingPolicy &rounding_policy) +{ + std::stringstream str; + str << rounding_policy; + return str.str(); +} + /** Formatted output of the DataType type. */ inline ::std::ostream &operator<<(::std::ostream &os, const DataType &data_type) { |