aboutsummaryrefslogtreecommitdiff
path: root/utils/TypePrinter.h
diff options
context:
space:
mode:
Diffstat (limited to 'utils/TypePrinter.h')
-rw-r--r--utils/TypePrinter.h18
1 files changed, 18 insertions, 0 deletions
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h
index d956d9ca01..50da59329a 100644
--- a/utils/TypePrinter.h
+++ b/utils/TypePrinter.h
@@ -151,6 +151,21 @@ inline ::std::ostream &operator<<(::std::ostream &os, const ROIPoolingLayerInfo
return os;
}
+/** Formatted output of the QuantizationInfo type. */
+inline ::std::ostream &operator<<(::std::ostream &os, const QuantizationInfo &quantization_info)
+{
+ os << "Scale:" << quantization_info.scale << "~"
+ << "Offset:" << quantization_info.offset;
+ return os;
+}
+
+inline std::string to_string(const QuantizationInfo &quantization_info)
+{
+ std::stringstream str;
+ str << quantization_info;
+ return str.str();
+}
+
inline ::std::ostream &operator<<(::std::ostream &os, const FixedPointOp &op)
{
switch(op)
@@ -333,6 +348,9 @@ inline ::std::ostream &operator<<(::std::ostream &os, const DataType &data_type)
case DataType::QS8:
os << "QS8";
break;
+ case DataType::QASYMM8:
+ os << "QASYMM8";
+ break;
case DataType::S8:
os << "S8";
break;