diff options
| author | Isabella Gottardi <isabella.gottardi@arm.com> | 2018-02-06 14:52:43 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:47:18 +0000 |
| commit | f07d28d9ee8ae73a93fe433f72855b6dcf58ad90 (patch) | |
| tree | 6ad19c89540f36e1ba5c6af7ff061bee773c43d6 /utils | |
| parent | 21f67d6763c82d78278f6bca6c6f9e42bb5ee1b9 (diff) | |
| download | ComputeLibrary-f07d28d9ee8ae73a93fe433f72855b6dcf58ad90.tar.gz | |
COMPMID-845: Create a ConvolutionLayer for CL
Change-Id: Ifcc406d2d0a99c911d6b6c875657b0e0028255d5
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/119148
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/TypePrinter.h | 66 |
1 files changed, 66 insertions, 0 deletions
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__ */ |