COMPMID-1367: Enable NHWC in graph examples
Change-Id: Iabc54a3a1bdcd46a9a921cda39c7c85fef672b72
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/141449
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h
index eb03ff9..8bfdba9 100644
--- a/utils/TypePrinter.h
+++ b/utils/TypePrinter.h
@@ -475,6 +475,35 @@
return str.str();
}
+/** Formatted output of the DataLayoutDimension type.
+ *
+ * @param[out] os Output stream.
+ * @param[in] data_layout_dim Data layout dimension to print.
+ *
+ * @return Modified output stream.
+ */
+inline ::std::ostream &operator<<(::std::ostream &os, const DataLayoutDimension &data_layout_dim)
+{
+ switch(data_layout_dim)
+ {
+ case DataLayoutDimension::WIDTH:
+ os << "WIDTH";
+ break;
+ case DataLayoutDimension::HEIGHT:
+ os << "HEIGHT";
+ break;
+ case DataLayoutDimension::CHANNEL:
+ os << "CHANNEL";
+ break;
+ case DataLayoutDimension::BATCHES:
+ os << "BATCHES";
+ break;
+ default:
+ ARM_COMPUTE_ERROR("NOT_SUPPORTED!");
+ }
+ return os;
+}
+
/** Formatted output of the DataType type.
*
* @param[out] os Output stream.