diff options
Diffstat (limited to 'utils/TypePrinter.h')
-rw-r--r-- | utils/TypePrinter.h | 19 |
1 files changed, 19 insertions, 0 deletions
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 |