diff options
Diffstat (limited to 'tests/TypePrinter.h')
-rw-r--r-- | tests/TypePrinter.h | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/tests/TypePrinter.h b/tests/TypePrinter.h index 020b559c6f..49e717a48a 100644 --- a/tests/TypePrinter.h +++ b/tests/TypePrinter.h @@ -86,6 +86,13 @@ inline ::std::ostream &operator<<(::std::ostream &os, const PadStrideInfo &pad_s return os; } +inline std::string to_string(const PadStrideInfo &pad_stride_info) +{ + std::stringstream str; + str << pad_stride_info; + return str.str(); +} + /** Formatted output of the ROIPoolingInfo type. */ inline ::std::ostream &operator<<(::std::ostream &os, const ROIPoolingLayerInfo &pool_info) { @@ -329,6 +336,13 @@ inline ::std::ostream &operator<<(::std::ostream &os, const PoolingType &pool_ty return os; } +inline std::string to_string(const PoolingType &type) +{ + std::stringstream str; + str << type; + return str.str(); +} + /** Formatted output of @ref PoolingLayerInfo. */ inline ::std::ostream &operator<<(::std::ostream &os, const PoolingLayerInfo &info) { @@ -337,6 +351,13 @@ inline ::std::ostream &operator<<(::std::ostream &os, const PoolingLayerInfo &in return os; } +inline std::string to_string(const PoolingLayerInfo &info) +{ + std::stringstream str; + str << info.pool_type(); + return str.str(); +} + /** Formatted output of the RoundingPolicy type. */ inline ::std::ostream &operator<<(::std::ostream &os, const RoundingPolicy &rounding_policy) { |