1 //
2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "RefDepthwiseConvolution2dWorkload.hpp"
7
8 #include "ConvImpl.hpp"
9 #include "RefWorkloadUtils.hpp"
10 #include "Decoders.hpp"
11 #include "Encoders.hpp"
12 #include "Profiling.hpp"
13 #include <ResolveType.hpp>
14
15 namespace armnn
16 {
17
RefDepthwiseConvolution2dWorkload(const DepthwiseConvolution2dQueueDescriptor & descriptor,const WorkloadInfo & info)18 RefDepthwiseConvolution2dWorkload::RefDepthwiseConvolution2dWorkload(
19 const DepthwiseConvolution2dQueueDescriptor& descriptor, const WorkloadInfo& info)
20 : RefBaseWorkload<DepthwiseConvolution2dQueueDescriptor>(descriptor, info)
21 {
22 WorkloadInfo detailsInfo;
23 detailsInfo.m_InputTensorInfos = info.m_InputTensorInfos;
24 detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos;
25 detailsInfo.m_WeightsTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[1]);
26
27 if (descriptor.m_Parameters.m_BiasEnabled)
28 {
29 detailsInfo.m_BiasTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[2]);
30 }
31
32 // Report Profiling Details
33 ARMNN_REPORT_PROFILING_WORKLOAD_DESC("RefDepthwiseConvolution2dWorkload_Construct",
34 descriptor.m_Parameters,
35 detailsInfo,
36 this->GetGuid());
37 }
38
Execute() const39 void RefDepthwiseConvolution2dWorkload::Execute() const
40 {
41 Execute(m_Data.m_Inputs, m_Data.m_Outputs);
42 }
43
ExecuteAsync(ExecutionData & executionData)44 void RefDepthwiseConvolution2dWorkload::ExecuteAsync(ExecutionData& executionData)
45 {
46 WorkingMemDescriptor* workingMemDescriptor = static_cast<WorkingMemDescriptor*>(executionData.m_Data);
47 Execute(workingMemDescriptor->m_Inputs, workingMemDescriptor->m_Outputs);
48 }
49
Execute(std::vector<ITensorHandle * > inputs,std::vector<ITensorHandle * > outputs) const50 void RefDepthwiseConvolution2dWorkload::Execute(std::vector<ITensorHandle*> inputs,
51 std::vector<ITensorHandle*> outputs) const
52 {
53 ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefDepthwiseConvolution2dWorkload_Execute");
54
55 const TensorShape& inputShape = GetTensorInfo(inputs[0]).GetShape();
56 const TensorShape& outputShape = GetTensorInfo(outputs[0]).GetShape();
57 const TensorShape& filterShape = GetTensorInfo(inputs[1]).GetShape();
58
59 std::unique_ptr<Decoder<float>> inputDecoder = MakeDecoder<float>(GetTensorInfo(inputs[0]), inputs[0]->Map());
60 std::unique_ptr<Encoder<float>> outputEncoder = MakeEncoder<float>(GetTensorInfo(outputs[0]), outputs[0]->Map());
61 std::unique_ptr<Decoder<float>> filterDecoder = MakeDecoder<float>(GetTensorInfo(inputs[1]), inputs[1]->Map());
62 std::unique_ptr<Decoder<float>> biasDecoder{};
63 if (m_Data.m_Parameters.m_BiasEnabled)
64 {
65 biasDecoder = MakeDecoder<float>(GetTensorInfo(inputs[2]), inputs[2]->Map());
66 }
67
68 Convolve(inputShape, *inputDecoder, outputShape, *outputEncoder,
69 filterShape, *filterDecoder, m_Data.m_Parameters.m_BiasEnabled, biasDecoder.get(),
70 m_Data.m_Parameters.m_DataLayout, m_Data.m_Parameters.m_PadTop, m_Data.m_Parameters.m_PadLeft,
71 m_Data.m_Parameters.m_StrideX, m_Data.m_Parameters.m_StrideY,
72 m_Data.m_Parameters.m_DilationX,
73 m_Data.m_Parameters.m_DilationY, true);
74 }
75
76 } //namespace armnn
77