COMPMID-1944 Add support for "reflect" padding mode in NEPad
Change-Id: I56c42524497d37d44708648571fa211ac1afbd98
Signed-off-by: Usama Arif <usama.arif@arm.com>
Reviewed-on: https://review.mlplatform.org/c/885
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h
index f2cf606..7c23399 100644
--- a/utils/TypePrinter.h
+++ b/utils/TypePrinter.h
@@ -1183,6 +1183,46 @@
return os;
}
+/** Formatted output of the PaddingMode type.
+ *
+ * @param[out] os Output stream.
+ * @param[in] mode Type to output.
+ *
+ * @return Modified output stream.
+ */
+inline ::std::ostream &operator<<(::std::ostream &os, const PaddingMode &mode)
+{
+ switch(mode)
+ {
+ case PaddingMode::CONSTANT:
+ os << "CONSTANT";
+ break;
+ case PaddingMode::REFLECT:
+ os << "REFLECT";
+ break;
+ case PaddingMode::SYMMETRIC:
+ os << "SYMMETRIC";
+ break;
+ default:
+ ARM_COMPUTE_ERROR("NOT_SUPPORTED!");
+ }
+
+ return os;
+}
+
+/** Formatted output of the PaddingMode type.
+ *
+ * @param[in] mode Type to output.
+ *
+ * @return Formatted string.
+ */
+inline std::string to_string(const PaddingMode &mode)
+{
+ std::stringstream str;
+ str << mode;
+ return str.str();
+}
+
/** Formatted output of the PadStrideInfo type.
*
* @param[out] os Output stream.