diff options
| author | kevmay01 <kevin.may@arm.com> | 2018-09-26 10:21:55 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:57 +0100 |
| commit | e448be3ac55897a3eabe85962891f8414f8e3cf9 (patch) | |
| tree | 11e46d4979eb6d8d70c7f43d5cf690fc6f78d605 /src/backends/NeonWorkloadFactory.cpp | |
| parent | 9fc824a596d6bddd27e5aa1438b115e71a117aa3 (diff) | |
| download | armnn-e448be3ac55897a3eabe85962891f8414f8e3cf9.tar.gz | |
IVGCVSW-1927 Add Neon 8-bit FullyConnected support
Change-Id: Idf4cc7a9a7d3261b9eceb653b999257506cdae76
Diffstat (limited to 'src/backends/NeonWorkloadFactory.cpp')
| -rw-r--r-- | src/backends/NeonWorkloadFactory.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/NeonWorkloadFactory.cpp b/src/backends/NeonWorkloadFactory.cpp index c989121eac..3b994bf049 100644 --- a/src/backends/NeonWorkloadFactory.cpp +++ b/src/backends/NeonWorkloadFactory.cpp @@ -116,8 +116,8 @@ std::unique_ptr<armnn::IWorkload> NeonWorkloadFactory::CreateMerger(const Merger std::unique_ptr<armnn::IWorkload> NeonWorkloadFactory::CreateFullyConnected( const FullyConnectedQueueDescriptor& descriptor, const WorkloadInfo& info) const { - return MakeWorkload<NeonFullyConnectedFloatWorkload, NullWorkload>(descriptor, info, - m_MemoryManager.GetIntraLayerManager()); + return MakeWorkload<NeonFullyConnectedWorkload, NeonFullyConnectedWorkload>(descriptor, info, + m_MemoryManager.GetIntraLayerManager()); } std::unique_ptr<armnn::IWorkload> NeonWorkloadFactory::CreatePermute(const PermuteQueueDescriptor& descriptor, |