diff options
Diffstat (limited to 'tests/TypePrinter.h')
| -rw-r--r-- | tests/TypePrinter.h | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/tests/TypePrinter.h b/tests/TypePrinter.h index d76754448e..f5915c474a 100644 --- a/tests/TypePrinter.h +++ b/tests/TypePrinter.h @@ -53,6 +53,18 @@ inline ::std::ostream &operator<<(::std::ostream &os, const Dimensions<T> &dimen return os; } +/** Formatted output of the Rectangle type. */ +inline ::std::ostream &operator<<(::std::ostream &os, const Rectangle &rect) +{ + os << "("; + os << rect.height << ", " << rect.width; + os << ", "; + os << rect.x << ", " << rect.y; + os << ")"; + + return os; +} + /** Formatted output of the PadStridInfo type. */ inline ::std::ostream &operator<<(::std::ostream &os, const PadStrideInfo &pad_stride_info) { @@ -65,6 +77,13 @@ inline ::std::ostream &operator<<(::std::ostream &os, const PadStrideInfo &pad_s return os; } +/** Formatted output of the ROIPoolingInfo type. */ +inline ::std::ostream &operator<<(::std::ostream &os, const ROIPoolingLayerInfo &pool_info) +{ + os << pool_info.pooled_width() << ", " << pool_info.pooled_height() << ", " << pool_info.spatial_scale(); + return os; +} + /** Formatted output of the BorderMode type. */ inline ::std::ostream &operator<<(::std::ostream &os, const BorderMode &mode) { |