1 //
2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "ClStridedSliceWorkload.hpp"
7
8 #include "ClWorkloadUtils.hpp"
9
10 #include <aclCommon/ArmComputeUtils.hpp>
11 #include <aclCommon/ArmComputeTensorUtils.hpp>
12
13 #include <armnn/backends/TensorHandle.hpp>
14 #include <backendsCommon/WorkloadUtils.hpp>
15
16 #include <armnn/utility/NumericCast.hpp>
17
18 #include <cl/ClLayerSupport.hpp>
19 #include <cl/ClTensorHandle.hpp>
20 #include <cl/ClLayerSupport.hpp>
21
22 namespace armnn
23 {
24
25 using namespace armcomputetensorutils;
26
ClStridedSliceWorkloadValidate(const TensorInfo & input,const TensorInfo & output,const StridedSliceDescriptor & descriptor)27 arm_compute::Status ClStridedSliceWorkloadValidate(const TensorInfo& input,
28 const TensorInfo& output,
29 const StridedSliceDescriptor& descriptor)
30 {
31 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout);
32 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout);
33
34 arm_compute::Coordinates starts;
35 arm_compute::Coordinates ends;
36 arm_compute::Coordinates strides;
37
38 std::tie(starts, ends, strides) = SetClStridedSliceData(descriptor.m_Begin, descriptor.m_End, descriptor.m_Stride);
39
40 auto numDimensions = armnn::numeric_cast<int>(input.GetNumDimensions());
41 int32_t begin_mask = ConvertMaskToACLFormat(descriptor.m_BeginMask, numDimensions);
42 int32_t end_mask = ConvertMaskToACLFormat(descriptor.m_EndMask, numDimensions);
43 int32_t shrink_axis_mask = ConvertMaskToACLFormat(descriptor.m_ShrinkAxisMask, numDimensions);
44
45 return arm_compute::CLStridedSlice::validate(&aclInputInfo,
46 &aclOutputInfo,
47 starts,
48 ends,
49 strides,
50 begin_mask,
51 end_mask,
52 shrink_axis_mask);
53 }
54
ClStridedSliceWorkload(const StridedSliceQueueDescriptor & descriptor,const WorkloadInfo & info,const arm_compute::CLCompileContext & clCompileContext)55 ClStridedSliceWorkload::ClStridedSliceWorkload(const StridedSliceQueueDescriptor& descriptor,
56 const WorkloadInfo& info,
57 const arm_compute::CLCompileContext& clCompileContext)
58 : ClBaseWorkload<StridedSliceQueueDescriptor>(descriptor, info)
59 {
60 // Report Profiling Details
61 ARMNN_REPORT_PROFILING_WORKLOAD_DESC("ClStridedSliceWorkload_Construct",
62 descriptor.m_Parameters,
63 info,
64 this->GetGuid());
65
66 m_Data.ValidateInputsOutputs("ClStridedSliceWorkload", 1, 1);
67
68 arm_compute::ICLTensor& input = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
69 arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
70
71 arm_compute::Coordinates starts;
72 arm_compute::Coordinates ends;
73 arm_compute::Coordinates strides;
74
75 std::tie(starts, ends, strides) = SetClStridedSliceData(m_Data.m_Parameters.m_Begin,
76 m_Data.m_Parameters.m_End,
77 m_Data.m_Parameters.m_Stride);
78
79 auto numDimensions = armnn::numeric_cast<int>(info.m_InputTensorInfos[0].GetNumDimensions());
80 int32_t begin_mask = ConvertMaskToACLFormat(m_Data.m_Parameters.m_BeginMask, numDimensions);
81 int32_t end_mask = ConvertMaskToACLFormat(m_Data.m_Parameters.m_EndMask, numDimensions);
82 int32_t shrink_axis_mask = ConvertMaskToACLFormat(m_Data.m_Parameters.m_ShrinkAxisMask, numDimensions);
83
84 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout);
85 input.info()->set_data_layout(aclDataLayout);
86 output.info()->set_data_layout(aclDataLayout);
87
88 {
89 ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "ClStridedSliceWorkload_configure");
90 m_StridedSliceLayer.configure(clCompileContext,
91 &input,
92 &output,
93 starts,
94 ends,
95 strides,
96 begin_mask,
97 end_mask,
98 shrink_axis_mask);
99 }
100 }
101
Execute() const102 void ClStridedSliceWorkload::Execute() const
103 {
104 ARMNN_SCOPED_PROFILING_EVENT_CL_GUID("ClStridedSliceWorkload_Execute", this->GetGuid());
105 RunClFunction(m_StridedSliceLayer, CHECK_LOCATION());
106 }
107
108 } //namespace armnn
109