COMPMID-1188 Added missing type printer for TensorInfo
Change-Id: Iba115d5df9d3b5802899318e2e68c33454731e33
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/142251
Reviewed-by: Michele DiGiorgio <michele.digiorgio@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h
index 8bfdba9..d9b971d 100644
--- a/utils/TypePrinter.h
+++ b/utils/TypePrinter.h
@@ -823,6 +823,20 @@
/** Formatted output of the TensorInfo type.
*
+ * @param[out] os Output stream.
+ * @param[in] info Type to output.
+ *
+ * @return Modified output stream.
+ */
+inline ::std::ostream &operator<<(::std::ostream &os, const TensorInfo &info)
+{
+ os << "{Shape=" << info.tensor_shape() << ","
+ << "Type=" << info.data_type() << ","
+ << "Channels=" << info.num_channels() << "}";
+ return os;
+}
+/** Formatted output of the TensorInfo type.
+ *
* @param[in] info Type to output.
*
* @return Formatted string.
@@ -830,9 +844,7 @@
inline std::string to_string(const TensorInfo &info)
{
std::stringstream str;
- str << "{Shape=" << info.tensor_shape() << ","
- << "Type=" << info.data_type() << ","
- << "Channels=" << info.num_channels() << "}";
+ str << info;
return str.str();
}