aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/workloads/NeonBaseConstantWorkload.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/neon/workloads/NeonBaseConstantWorkload.hpp')
-rw-r--r--src/backends/neon/workloads/NeonBaseConstantWorkload.hpp82
1 files changed, 0 insertions, 82 deletions
diff --git a/src/backends/neon/workloads/NeonBaseConstantWorkload.hpp b/src/backends/neon/workloads/NeonBaseConstantWorkload.hpp
deleted file mode 100644
index 828e476d29..0000000000
--- a/src/backends/neon/workloads/NeonBaseConstantWorkload.hpp
+++ /dev/null
@@ -1,82 +0,0 @@
-//
-// Copyright © 2017 Arm Ltd. All rights reserved.
-// SPDX-License-Identifier: MIT
-//
-
-#pragma once
-
-#include <arm_compute/core/Types.h>
-#include <armnnUtils/Half.hpp>
-#include <backends/aclCommon/ArmComputeTensorUtils.hpp>
-#include <backends/neon/NeonTensorHandle.hpp>
-#include <backends/neon/workloads/NeonWorkloadUtils.hpp>
-#include <backends/CpuTensorHandle.hpp>
-#include <backends/Workload.hpp>
-
-#include <boost/cast.hpp>
-
-namespace armnn
-{
-
-// Base class template providing an implementation of the Constant layer common to all data types.
-template <armnn::DataType... DataFormats>
-class NeonBaseConstantWorkload : public TypedWorkload<ConstantQueueDescriptor, DataFormats...>
-{
-public:
- NeonBaseConstantWorkload(const ConstantQueueDescriptor& descriptor, const WorkloadInfo& info)
- : TypedWorkload<ConstantQueueDescriptor, DataFormats...>(descriptor, info)
- , m_RanOnce(false)
- {
- }
-
- virtual void Execute() const override
- {
- using namespace armcomputetensorutils;
-
- // The intermediate tensor held by the corresponding layer output handler can be initialised with the
- // given data on the first inference, then reused for subsequent inferences.
- // The initialisation cannot happen at workload construction time since the ACL kernel for the next layer
- // may not have been configured at the time.
- if (!m_RanOnce)
- {
- const ConstantQueueDescriptor& data = this->m_Data;
-
- BOOST_ASSERT(data.m_LayerOutput != nullptr);
- arm_compute::ITensor& output =
- boost::polymorphic_downcast<NeonTensorHandle*>(data.m_Outputs[0])->GetTensor();
- arm_compute::DataType computeDataType =
- boost::polymorphic_downcast<NeonTensorHandle*>(data.m_Outputs[0])->GetDataType();
-
- switch (computeDataType)
- {
- case arm_compute::DataType::F16:
- {
- CopyArmComputeITensorData(data.m_LayerOutput->GetConstTensor<Half>(), output);
- break;
- }
- case arm_compute::DataType::F32:
- {
- CopyArmComputeITensorData(data.m_LayerOutput->GetConstTensor<float>(), output);
- break;
- }
- case arm_compute::DataType::QASYMM8:
- {
- CopyArmComputeITensorData(data.m_LayerOutput->GetConstTensor<uint8_t>(), output);
- break;
- }
- default:
- {
- BOOST_ASSERT_MSG(false, "Unknown data type");
- break;
- }
- }
-
- m_RanOnce = true;
- }
- }
-
-private:
- mutable bool m_RanOnce;
-};
-
-} //namespace armnn