diff options
author | Matthew Bentham <Matthew.Bentham@arm.com> | 2020-02-05 21:39:55 +0000 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2020-02-10 13:29:08 +0000 |
commit | 9b3e738929b2fc15a1a79b36b21267790ee34cb5 (patch) | |
tree | c9dcb1af167c8d8bc493cf7227d0e234af615e65 /src/backends/cl/workloads/ClSplitterWorkload.cpp | |
parent | 416b41da63ec61bddd1e642aac90c88318269749 (diff) | |
download | armnn-9b3e738929b2fc15a1a79b36b21267790ee34cb5.tar.gz |
Clean up header inclusion in CL workloads.
Remove usage of CLFunctions.h and replace with specific header
for each function.
Signed-off-by: Matthew Bentham <Matthew.Bentham@arm.com>
Change-Id: I255a843b36a5dd8cf90a52786282192b8fe04f83
Diffstat (limited to 'src/backends/cl/workloads/ClSplitterWorkload.cpp')
-rw-r--r-- | src/backends/cl/workloads/ClSplitterWorkload.cpp | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/src/backends/cl/workloads/ClSplitterWorkload.cpp b/src/backends/cl/workloads/ClSplitterWorkload.cpp index 9bbbcab797..296e0a3dde 100644 --- a/src/backends/cl/workloads/ClSplitterWorkload.cpp +++ b/src/backends/cl/workloads/ClSplitterWorkload.cpp @@ -9,6 +9,7 @@ #include <aclCommon/ArmComputeTensorUtils.hpp> #include <aclCommon/ArmComputeUtils.hpp> +#include <arm_compute/runtime/CL/functions/CLSplit.h> #include <backendsCommon/CpuTensorHandle.hpp> #include <cl/ClTensorHandle.hpp> @@ -84,7 +85,6 @@ ClSplitterWorkload::ClSplitterWorkload(const SplitterQueueDescriptor& descriptor } // Create the layer function - m_Layer.reset(new arm_compute::CLSplit()); // Configure input and output tensors std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor.m_Parameters, m_Data.m_Inputs[0]->GetShape()); @@ -94,10 +94,13 @@ ClSplitterWorkload::ClSplitterWorkload(const SplitterQueueDescriptor& descriptor } unsigned int aclAxis = CalcAclAxis(descriptor.m_Parameters.GetNumDimensions(), *splitAxis.begin()); - m_Layer->configure(&input, aclOutputs, aclAxis); + auto layer = std::make_unique<arm_compute::CLSplit>(); + layer->configure(&input, aclOutputs, aclAxis); // Prepare - m_Layer->prepare(); + layer->prepare(); + + m_Layer = std::move(layer); } void ClSplitterWorkload::Execute() const |