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>
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h
index 52699b6..63fba35 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 @@
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__ */