diff options
author | Matthew Bentham <matthew.bentham@arm.com> | 2018-09-17 11:17:41 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:56 +0100 |
commit | ab8cdc13443408d727fd38be42315cf942251940 (patch) | |
tree | f1576377150b72f1a2404915adcf1affd5eab3e4 /src/armnn/backends/ClWorkloadFactory.cpp | |
parent | ca225f0ab9aac74ccc7c62cfcf46c95f7715b2ee (diff) | |
download | armnn-ab8cdc13443408d727fd38be42315cf942251940.tar.gz |
IVGCVSW-949 Add 8-bit fully connected support
Change-Id: I0953bb8dbc4b76001f207e37c8c2742a6ebd888b
Diffstat (limited to 'src/armnn/backends/ClWorkloadFactory.cpp')
-rw-r--r-- | src/armnn/backends/ClWorkloadFactory.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnn/backends/ClWorkloadFactory.cpp b/src/armnn/backends/ClWorkloadFactory.cpp index c35f044e9e..591fb85dbb 100644 --- a/src/armnn/backends/ClWorkloadFactory.cpp +++ b/src/armnn/backends/ClWorkloadFactory.cpp @@ -116,8 +116,8 @@ std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreateMerger(const MergerQu std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreateFullyConnected( const FullyConnectedQueueDescriptor& descriptor, const WorkloadInfo& info) const { - return MakeWorkload<ClFullyConnectedFloatWorkload, NullWorkload>(descriptor, info, - m_MemoryManager.GetIntraLayerManager()); + return MakeWorkload<ClFullyConnectedWorkload, ClFullyConnectedWorkload>(descriptor, info, + m_MemoryManager.GetIntraLayerManager()); } std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreatePermute(const PermuteQueueDescriptor& descriptor, |