COMPMID-1013 - Create WinogradInfo data structure
COMPMID-1014 - Refactoring Winograd's dataset
Change-Id: I6abdcbf9a90d663f4db666cd410afece9f1d034d
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/125899
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h
index 9ac4b34..e26b695 100644
--- a/utils/TypePrinter.h
+++ b/utils/TypePrinter.h
@@ -1457,6 +1457,7 @@
str << gpu_target;
return str.str();
}
+
/** Formatted output of the DetectionWindow type. */
inline ::std::ostream &operator<<(::std::ostream &os, const DetectionWindow &detection_window)
{
@@ -1470,11 +1471,29 @@
return os;
}
+/** Formatted output of the WinogradInfo type. */
+inline ::std::ostream &operator<<(::std::ostream &os, const WinogradInfo &info)
+{
+ os << "{OutputTileSize=" << info.output_tile_size << ","
+ << "KernelSize=" << info.kernel_size << ","
+ << "PadStride=" << info.convolution_info << ","
+ << "OutputDataLayout=" << info.output_data_layout << "}";
+
+ return os;
+}
+
inline std::string to_string(const DetectionWindow &type)
{
std::stringstream str;
str << type;
return str.str();
}
+
+inline std::string to_string(const WinogradInfo &type)
+{
+ std::stringstream str;
+ str << type;
+ return str.str();
+}
} // namespace arm_compute
#endif /* __ARM_COMPUTE_TEST_TYPE_PRINTER_H__ */
\ No newline at end of file