diff options
author | Ruomei Yan <ruomei.yan@arm.com> | 2019-05-28 16:48:20 +0100 |
---|---|---|
committer | Ruomei Yan <ruomei.yan@arm.com> | 2019-05-30 15:45:40 +0100 |
commit | 25339c31a829111fce691311cb84100e8591f5da (patch) | |
tree | 9d79436f67f4570e4e4d1c52516de7f3df6eb7e0 /src/backends/reference/RefWorkloadFactory.cpp | |
parent | edb8b2ec42624ec27c37bc1b3d345ef0c97c024c (diff) | |
download | armnn-25339c31a829111fce691311cb84100e8591f5da.tar.gz |
IVGCVSW-3159 Support QSymm16 for Splitter workloads
Change-Id: I9af5d2d8ade97b9ecd2e6fbf13db9fa3bb622ed8
Signed-off-by: Ruomei Yan <ruomei.yan@arm.com>
Diffstat (limited to 'src/backends/reference/RefWorkloadFactory.cpp')
-rw-r--r-- | src/backends/reference/RefWorkloadFactory.cpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/backends/reference/RefWorkloadFactory.cpp b/src/backends/reference/RefWorkloadFactory.cpp index 50e3c0006c..71b1cb034d 100644 --- a/src/backends/reference/RefWorkloadFactory.cpp +++ b/src/backends/reference/RefWorkloadFactory.cpp @@ -133,7 +133,11 @@ std::unique_ptr<IWorkload> RefWorkloadFactory::CreateSoftmax(const SoftmaxQueueD std::unique_ptr<IWorkload> RefWorkloadFactory::CreateSplitter(const SplitterQueueDescriptor& descriptor, const WorkloadInfo& info) const { - return MakeWorkload<RefSplitterFloat32Workload, RefSplitterUint8Workload>(descriptor, info); + if (IsFloat16(info)) + { + return MakeWorkload<NullWorkload, NullWorkload>(descriptor, info); + } + return std::make_unique<RefSplitterWorkload>(descriptor, info); } std::unique_ptr<armnn::IWorkload> RefWorkloadFactory::CreateMerger(const MergerQueueDescriptor& descriptor, |