From 5a64f22101ecdda4846e9d71428633f3ccd56fb2 Mon Sep 17 00:00:00 2001 From: Keith Davis Date: Wed, 4 Aug 2021 10:35:20 +0100 Subject: IVGCVSW-5980 Add Descriptor, TensorInfo and Convolution algorithm to JSON * Add GUID as field to layer details and profiling events * Add Optional GUID param to existing tests * Improve Details macro to be inline function * Fix some formatting Signed-off-by: Keith Davis Change-Id: I66f192a90a7642b3ee8e7dda0d3f428cce002581 --- src/backends/neon/workloads/NeonWorkloadUtils.hpp | 25 +++++++++++++++++++++++ 1 file changed, 25 insertions(+) (limited to 'src/backends/neon/workloads/NeonWorkloadUtils.hpp') diff --git a/src/backends/neon/workloads/NeonWorkloadUtils.hpp b/src/backends/neon/workloads/NeonWorkloadUtils.hpp index ab7616fbe2..1199f30863 100644 --- a/src/backends/neon/workloads/NeonWorkloadUtils.hpp +++ b/src/backends/neon/workloads/NeonWorkloadUtils.hpp @@ -16,6 +16,14 @@ #define ARMNN_SCOPED_PROFILING_EVENT_NEON(name) \ ARMNN_SCOPED_PROFILING_EVENT_WITH_INSTRUMENTS(armnn::Compute::CpuAcc, \ + armnn::EmptyOptional(), \ + name, \ + armnn::NeonTimer(), \ + armnn::WallClockTimer()) + +#define ARMNN_SCOPED_PROFILING_EVENT_NEON_GUID(name, guid) \ + ARMNN_SCOPED_PROFILING_EVENT_WITH_INSTRUMENTS(armnn::Compute::CpuAcc, \ + guid, \ name, \ armnn::NeonTimer(), \ armnn::WallClockTimer()) @@ -25,6 +33,23 @@ using namespace armnn::armcomputetensorutils; namespace armnn { +inline std::string GetConvolutionMethodString(arm_compute::ConvolutionMethod& convolutionMethod) +{ + switch (convolutionMethod) + { + case arm_compute::ConvolutionMethod::FFT: + return "FFT"; + case arm_compute::ConvolutionMethod::DIRECT: + return "Direct"; + case arm_compute::ConvolutionMethod::GEMM: + return "GEMM"; + case arm_compute::ConvolutionMethod::WINOGRAD: + return "Winograd"; + default: + return "Unknown"; + } +} + template void CopyArmComputeTensorData(arm_compute::Tensor& dstTensor, const T* srcData) { -- cgit v1.2.1