1 //
2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "ClSpaceToDepthWorkload.hpp"
7 #include "ClWorkloadUtils.hpp"
8
9 #include <aclCommon/ArmComputeUtils.hpp>
10 #include <aclCommon/ArmComputeTensorUtils.hpp>
11 #include <armnn/backends/TensorHandle.hpp>
12 #include <cl/ClTensorHandle.hpp>
13
14 #include <armnn/utility/NumericCast.hpp>
15
16 namespace armnn
17 {
18 using namespace armcomputetensorutils;
19
ClSpaceToDepthWorkload(const SpaceToDepthQueueDescriptor & descriptor,const WorkloadInfo & info,const arm_compute::CLCompileContext & clCompileContext)20 ClSpaceToDepthWorkload::ClSpaceToDepthWorkload(const SpaceToDepthQueueDescriptor& descriptor,
21 const WorkloadInfo& info,
22 const arm_compute::CLCompileContext& clCompileContext)
23 : ClBaseWorkload<SpaceToDepthQueueDescriptor>(descriptor, info)
24 {
25 // Report Profiling Details
26 ARMNN_REPORT_PROFILING_WORKLOAD_DESC("ClSpaceToDepthWorkload_Construct",
27 descriptor.m_Parameters,
28 info,
29 this->GetGuid());
30 m_Data.ValidateInputsOutputs("ClSpaceToDepthWorkload", 1, 1);
31
32 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout);
33
34 arm_compute::ICLTensor& input = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
35 input.info()->set_data_layout(aclDataLayout);
36
37 int32_t blockSize = armnn::numeric_cast<int32_t>(descriptor.m_Parameters.m_BlockSize);
38
39 arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
40 output.info()->set_data_layout(aclDataLayout);
41
42 {
43 ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "ClSpaceToDepthWorkload_configure");
44 m_Layer.configure(clCompileContext, &input, &output, blockSize);
45 }
46 }
47
Execute() const48 void ClSpaceToDepthWorkload::Execute() const
49 {
50 ARMNN_SCOPED_PROFILING_EVENT_CL_GUID("ClSpaceToDepthWorkload_Execute", this->GetGuid());
51 RunClFunction(m_Layer, CHECK_LOCATION());
52 }
53
ClSpaceToDepthWorkloadValidate(const TensorInfo & input,const TensorInfo & output,const SpaceToDepthDescriptor & descriptor)54 arm_compute::Status ClSpaceToDepthWorkloadValidate(const TensorInfo& input,
55 const TensorInfo& output,
56 const SpaceToDepthDescriptor& descriptor)
57 {
58 DataLayout dataLayout = descriptor.m_DataLayout;
59 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, dataLayout);
60
61 int32_t blockSize = armnn::numeric_cast<int32_t>(descriptor.m_BlockSize);
62
63 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, dataLayout);
64
65 const arm_compute::Status aclStatus = arm_compute::CLSpaceToDepthLayer::validate(&aclInputInfo,
66 &aclOutputInfo,
67 blockSize);
68 return aclStatus;
69 }
70
71 } //namespace armnn
72