From 247f52cfe337f7b2542b900e3d8cf122e9d4f11c Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Thu, 22 Mar 2018 11:24:56 +0000 Subject: 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 Reviewed-by: Anthony Barbier --- utils/TypePrinter.h | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'utils') diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h index 9ac4b343b3..e26b695809 100644 --- a/utils/TypePrinter.h +++ b/utils/TypePrinter.h @@ -1457,6 +1457,7 @@ inline std::string to_string(const GPUTarget &gpu_target) 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 @@ inline ::std::ostream &operator<<(::std::ostream &os, const DetectionWindow &det 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 -- cgit v1.2.1