From f07d28d9ee8ae73a93fe433f72855b6dcf58ad90 Mon Sep 17 00:00:00 2001 From: Isabella Gottardi Date: Tue, 6 Feb 2018 14:52:43 +0000 Subject: COMPMID-845: Create a ConvolutionLayer for CL Change-Id: Ifcc406d2d0a99c911d6b6c875657b0e0028255d5 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/119148 Tested-by: Jenkins Reviewed-by: Anthony Barbier Reviewed-by: Georgios Pinitas --- utils/TypePrinter.h | 66 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 66 insertions(+) (limited to 'utils') diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h index 52699b67de..63fba35052 100644 --- a/utils/TypePrinter.h +++ b/utils/TypePrinter.h @@ -24,6 +24,7 @@ #ifndef __ARM_COMPUTE_TEST_TYPE_PRINTER_H__ #define __ARM_COMPUTE_TEST_TYPE_PRINTER_H__ +#include "arm_compute/core/CL/CLTypes.h" #include "arm_compute/core/Dimensions.h" #include "arm_compute/core/Error.h" #include "arm_compute/core/HOGInfo.h" @@ -932,5 +933,70 @@ inline std::string to_string(const HOGInfo &type) return str.str(); } +inline ::std::ostream &operator<<(::std::ostream &os, const ConvolutionMethod &conv_method) +{ + switch(conv_method) + { + case ConvolutionMethod::GEMM: + os << "GEMM"; + break; + case ConvolutionMethod::DIRECT: + os << "DIRECT"; + break; + case ConvolutionMethod::WINOGRAD: + os << "WINOGRAD"; + break; + default: + ARM_COMPUTE_ERROR("NOT_SUPPORTED!"); + } + + return os; +} + +inline std::string to_string(const ConvolutionMethod &conv_method) +{ + std::stringstream str; + str << conv_method; + return str.str(); +} + +inline ::std::ostream &operator<<(::std::ostream &os, const GPUTarget &gpu_target) +{ + switch(gpu_target) + { + case GPUTarget::GPU_ARCH_MASK: + os << "GPU_ARCH_MASK"; + break; + case GPUTarget::MIDGARD: + os << "MIDGARD"; + break; + case GPUTarget::BIFROST: + os << "BIFROST"; + break; + case GPUTarget::T600: + os << "T600"; + break; + case GPUTarget::T700: + os << "T700"; + break; + case GPUTarget::T800: + os << "T800"; + break; + case GPUTarget::G70: + os << "G70"; + break; + default: + ARM_COMPUTE_ERROR("NOT_SUPPORTED!"); + } + + return os; +} + +inline std::string to_string(const GPUTarget &gpu_target) +{ + std::stringstream str; + str << gpu_target; + return str.str(); +} } // namespace arm_compute #endif /* __ARM_COMPUTE_TEST_TYPE_PRINTER_H__ */ -- cgit v1.2.1