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/ClConcatWorkload.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/ClConcatWorkload.cpp')
-rw-r--r-- | src/backends/cl/workloads/ClConcatWorkload.cpp | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/src/backends/cl/workloads/ClConcatWorkload.cpp b/src/backends/cl/workloads/ClConcatWorkload.cpp index fb28946549..5370466163 100644 --- a/src/backends/cl/workloads/ClConcatWorkload.cpp +++ b/src/backends/cl/workloads/ClConcatWorkload.cpp @@ -10,6 +10,7 @@ #include <cl/ClLayerSupport.hpp> #include <arm_compute/core/Types.h> +#include <arm_compute/runtime/CL/functions/CLConcatenateLayer.h> #include <boost/polymorphic_pointer_cast.hpp> @@ -78,14 +79,15 @@ ClConcatWorkload::ClConcatWorkload(const ConcatQueueDescriptor& descriptor, cons m_Data.m_Outputs[0])->GetTensor(); // Create the layer function - m_Layer.reset(new arm_compute::CLConcatenateLayer()); + auto layer = std::make_unique<arm_compute::CLConcatenateLayer>(); // Configure input and output tensors size_t aclAxis = CalcAxis(descriptor.m_Parameters); - m_Layer->configure(aclInputs, &output, aclAxis); + layer->configure(aclInputs, &output, aclAxis); // Prepare - m_Layer->prepare(); + layer->prepare(); + m_Layer = std::move(layer); } void ClConcatWorkload::Execute() const @@ -97,4 +99,4 @@ void ClConcatWorkload::Execute() const } } -} //namespace armnn
\ No newline at end of file +} //namespace armnn |