diff options
author | Pablo Tello <pablo.tello@arm.com> | 2020-01-23 10:37:17 +0000 |
---|---|---|
committer | Pablo Tello <pablo.tello@arm.com> | 2020-01-24 14:50:49 +0000 |
commit | e61f0712dc480f70372fcfe0921a85a7de53661b (patch) | |
tree | a6a87887b8269a104b898a226c2d7cc31433de21 /src/backends/neon/NeonWorkloadFactory.cpp | |
parent | 0be3a88518a73819aaf71ef71f4fb2fbfc14bc88 (diff) | |
download | armnn-e61f0712dc480f70372fcfe0921a85a7de53661b.tar.gz |
MLCE-156: Add Division workload in CpuAcc
Change-Id: I1f228fcaf1077867d9755a2b850c6703387fab34
Signed-off-by: Pablo Tello <pablo.tello@arm.com>
Diffstat (limited to 'src/backends/neon/NeonWorkloadFactory.cpp')
-rw-r--r-- | src/backends/neon/NeonWorkloadFactory.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/neon/NeonWorkloadFactory.cpp b/src/backends/neon/NeonWorkloadFactory.cpp index 649cb9f699..cb2e88eddb 100644 --- a/src/backends/neon/NeonWorkloadFactory.cpp +++ b/src/backends/neon/NeonWorkloadFactory.cpp @@ -216,10 +216,10 @@ std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateDetectionPostProcess( std::unique_ptr<armnn::IWorkload> NeonWorkloadFactory::CreateDivision( const DivisionQueueDescriptor& descriptor, const WorkloadInfo& info) const { - return MakeWorkloadHelper<NullWorkload, NullWorkload>(descriptor, info); + return std::make_unique<NeonDivisionWorkload>(descriptor, info); } -std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateElementwiseUnary(const ElementwiseUnaryQueueDescriptor& +std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateElementwiseUnary(const ElementwiseUnaryQueueDescriptor& descriptor, const WorkloadInfo& info) const { |