diff options
author | Jim Flynn <jim.flynn@arm.com> | 2020-03-19 17:03:14 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2020-03-19 17:03:14 +0000 |
commit | 0e2bab81442ee6cc2b94e4f7881ed0c5c6af65e7 (patch) | |
tree | b0af08b5a0b74149fca422151127ac6310385399 /src/backends/cl/workloads/ClSpaceToBatchNdWorkload.cpp | |
parent | 8c3259fa007d43fcc5ea56fe6928526dbe79f3c0 (diff) | |
download | armnn-0e2bab81442ee6cc2b94e4f7881ed0c5c6af65e7.tar.gz |
Creating gh-pages documentation for ArmNN
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'src/backends/cl/workloads/ClSpaceToBatchNdWorkload.cpp')
-rw-r--r-- | src/backends/cl/workloads/ClSpaceToBatchNdWorkload.cpp | 85 |
1 files changed, 0 insertions, 85 deletions
diff --git a/src/backends/cl/workloads/ClSpaceToBatchNdWorkload.cpp b/src/backends/cl/workloads/ClSpaceToBatchNdWorkload.cpp deleted file mode 100644 index 64da92c815..0000000000 --- a/src/backends/cl/workloads/ClSpaceToBatchNdWorkload.cpp +++ /dev/null @@ -1,85 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#include "ClSpaceToBatchNdWorkload.hpp" - -#include "ClWorkloadUtils.hpp" - -#include <aclCommon/ArmComputeUtils.hpp> -#include <aclCommon/ArmComputeTensorUtils.hpp> -#include <backendsCommon/CpuTensorHandle.hpp> -#include <cl/ClLayerSupport.hpp> -#include <cl/ClTensorHandle.hpp> -#include <cl/ClLayerSupport.hpp> - -#include <boost/polymorphic_pointer_cast.hpp> - -namespace armnn -{ -using namespace armcomputetensorutils; - -arm_compute::Status ClSpaceToBatchNdWorkloadValidate(const TensorInfo& input, - const TensorInfo& output, - const SpaceToBatchNdDescriptor& descriptor) -{ - const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); - const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); - - // ArmNN blockShape is [H, W] Cl asks for W, H - int32_t blockHeight = boost::numeric_cast<int32_t>(descriptor.m_BlockShape[0]); - int32_t blockWidth = boost::numeric_cast<int32_t>(descriptor.m_BlockShape[1]); - - arm_compute::Size2D paddingLeftTop = BuildArmComputeSize2D( - descriptor.m_PadList[1].first, descriptor.m_PadList[0].first); - arm_compute::Size2D paddingRightBottom = BuildArmComputeSize2D( - descriptor.m_PadList[1].second, descriptor.m_PadList[0].second); - - return arm_compute::CLSpaceToBatchLayer::validate(&aclInputInfo, - blockWidth, - blockHeight, - paddingLeftTop, - paddingRightBottom, - &aclOutputInfo); -} - -ClSpaceToBatchNdWorkload::ClSpaceToBatchNdWorkload( - const SpaceToBatchNdQueueDescriptor& descriptor, const WorkloadInfo& info) - : BaseWorkload<SpaceToBatchNdQueueDescriptor>(descriptor, info) -{ - m_Data.ValidateInputsOutputs("ClSpaceToBatchNdWorkload", 1, 1); - - arm_compute::ICLTensor& input = - boost::polymorphic_pointer_downcast<IClTensorHandle>(m_Data.m_Inputs[0])->GetTensor(); - arm_compute::ICLTensor& output = - boost::polymorphic_pointer_downcast<IClTensorHandle>(m_Data.m_Outputs[0])->GetTensor(); - - // ArmNN blockShape is [H, W] Cl asks for W, H - int32_t blockHeight = boost::numeric_cast<int32_t>(m_Data.m_Parameters.m_BlockShape[0]); - int32_t blockWidth = boost::numeric_cast<int32_t>(m_Data.m_Parameters.m_BlockShape[1]); - - arm_compute::Size2D paddingLeftTop = BuildArmComputeSize2D( - m_Data.m_Parameters.m_PadList[1].first, m_Data.m_Parameters.m_PadList[0].first); - arm_compute::Size2D paddingRightBottom = BuildArmComputeSize2D( - m_Data.m_Parameters.m_PadList[1].second, m_Data.m_Parameters.m_PadList[0].second); - - arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); - input.info()->set_data_layout(aclDataLayout); - output.info()->set_data_layout(aclDataLayout); - - m_SpaceToBatchLayer.configure(&input, - blockWidth, - blockHeight, - paddingLeftTop, - paddingRightBottom, - &output); -} - -void ClSpaceToBatchNdWorkload::Execute() const -{ - ARMNN_SCOPED_PROFILING_EVENT_CL("ClSpaceToBatchNdWorkload_Execute"); - RunClFunction(m_SpaceToBatchLayer, CHECK_LOCATION()); -} - -} //namespace armnn |