COMPMID-671: Add global pooling layer support.
Change-Id: Iead7497cc03e1e7bde440d2965a7bf54cbfa88bf
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/95579
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Joel Liang <joel.liang@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h
index 2d5c618..863fcaf 100644
--- a/utils/TypePrinter.h
+++ b/utils/TypePrinter.h
@@ -727,7 +727,15 @@
inline std::string to_string(const PoolingLayerInfo &info)
{
std::stringstream str;
- str << info.pool_type();
+ str << "{Type=" << info.pool_type() << ","
+ << "IsGlobalPooling=" << info.is_global_pooling();
+ if(!info.is_global_pooling())
+ {
+ str << ","
+ << "PoolSize=" << info.pool_size() << ","
+ << "PadStride=" << info.pad_stride_info();
+ }
+ str << "}";
return str.str();
}