diff options
| author | Keith Davis <keith.davis@arm.com> | 2021-08-05 11:35:00 +0100 |
|---|---|---|
| committer | KeithARM <keith.davis@arm.com> | 2021-08-10 11:53:19 +0000 |
| commit | 2d0679f33f75c43e7169fe0f0ee2d15d0620e091 (patch) | |
| tree | 9f259a41e3208aa37a19330b63e0aabac607cacf /src/backends/neon/workloads/NeonTransposeConvolution2dWorkload.cpp | |
| parent | d218d9804723e78da9bbd36e6211b3310426852b (diff) | |
| download | armnn-2d0679f33f75c43e7169fe0f0ee2d15d0620e091.tar.gz | |
IVGCVSW-6249 Add ProfilingDetails Macros to all workloads in Neon
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: I7be77712a9f790928219ce91222d46cc766ab9dd
Diffstat (limited to 'src/backends/neon/workloads/NeonTransposeConvolution2dWorkload.cpp')
| -rw-r--r-- | src/backends/neon/workloads/NeonTransposeConvolution2dWorkload.cpp | 19 |
1 files changed, 18 insertions, 1 deletions
diff --git a/src/backends/neon/workloads/NeonTransposeConvolution2dWorkload.cpp b/src/backends/neon/workloads/NeonTransposeConvolution2dWorkload.cpp index a1e545ce05..f9e1b36306 100644 --- a/src/backends/neon/workloads/NeonTransposeConvolution2dWorkload.cpp +++ b/src/backends/neon/workloads/NeonTransposeConvolution2dWorkload.cpp @@ -77,6 +77,23 @@ NeonTransposeConvolution2dWorkload::NeonTransposeConvolution2dWorkload( arm_compute::PadStrideInfo padStrideInfo = BuildArmComputePadStrideInfo(m_Data.m_Parameters); + // Add details for profiling output + WorkloadInfo detailsInfo; + + detailsInfo.m_InputTensorInfos = info.m_InputTensorInfos; + detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos; + detailsInfo.m_WeightsTensorInfo = armnn::Optional<armnn::TensorInfo>(descriptor.m_Weight->GetTensorInfo()); + if (descriptor.m_Parameters.m_BiasEnabled) + { + detailsInfo.m_BiasTensorInfo = armnn::Optional<armnn::TensorInfo>(descriptor.m_Bias->GetTensorInfo()); + } + + // Report Profiling Details + ARMNN_REPORT_PROFILING_WORKLOAD_DESC("NeonTransposeConvolution2dWorkload_Construct", + descriptor.m_Parameters, + detailsInfo, + this->GetGuid()); + m_Layer = std::make_unique<arm_compute::NEDeconvolutionLayer>(memoryManager); m_Layer->configure(&input, m_KernelTensor.get(), m_BiasTensor.get(), &output, padStrideInfo); @@ -95,7 +112,7 @@ NeonTransposeConvolution2dWorkload::NeonTransposeConvolution2dWorkload( void NeonTransposeConvolution2dWorkload::Execute() const { - ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonTransposeConvolution2dWorkload_Execute"); + ARMNN_SCOPED_PROFILING_EVENT_NEON_GUID("NeonTransposeConvolution2dWorkload_Execute", this->GetGuid()); m_Layer->run(); } |