diff options
Diffstat (limited to 'src/backends/neon/workloads/NeonSubtractionFloatWorkload.cpp')
-rw-r--r-- | src/backends/neon/workloads/NeonSubtractionFloatWorkload.cpp | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/src/backends/neon/workloads/NeonSubtractionFloatWorkload.cpp b/src/backends/neon/workloads/NeonSubtractionFloatWorkload.cpp index 1eae0a49ce..e39f8aa18e 100644 --- a/src/backends/neon/workloads/NeonSubtractionFloatWorkload.cpp +++ b/src/backends/neon/workloads/NeonSubtractionFloatWorkload.cpp @@ -4,9 +4,13 @@ // #include "NeonSubtractionFloatWorkload.hpp" + +#include "NeonWorkloadUtils.hpp" #include <aclCommon/ArmComputeTensorUtils.hpp> #include <backendsCommon/CpuTensorHandle.hpp> +#include <arm_compute/runtime/NEON/functions/NEArithmeticSubtraction.h> + namespace armnn { @@ -34,13 +38,15 @@ NeonSubtractionFloatWorkload::NeonSubtractionFloatWorkload(const SubtractionQueu arm_compute::ITensor& input2 = boost::polymorphic_downcast<INeonTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); arm_compute::ITensor& output = boost::polymorphic_downcast<INeonTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); - m_SubLayer.configure(&input1, &input2, &output, arm_compute::ConvertPolicy::SATURATE); + auto layer = std::make_unique<arm_compute::NEArithmeticSubtraction>(); + layer->configure(&input1, &input2, &output, arm_compute::ConvertPolicy::SATURATE); + m_SubLayer.reset(layer.release()); } void NeonSubtractionFloatWorkload::Execute() const { ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonSubtractionFloatWorkload_Execute"); - m_SubLayer.run(); + m_SubLayer->run(); } } //namespace armnn |