diff options
| author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-04-24 16:19:57 +0100 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-04-29 11:04:20 +0000 |
| commit | ae2c5f0350a7033f58578f9c509345445a639865 (patch) | |
| tree | 4fd6266fc7d917130a0087938eafcadd1b7f31be /src/backends/reference/RefWorkloadFactory.cpp | |
| parent | bb73e576e75fde75acb28362fb7be194dfcbcada (diff) | |
| download | armnn-ae2c5f0350a7033f58578f9c509345445a639865.tar.gz | |
IVGCVSW-2982 Refactor reference Activation workload
Change-Id: Ia3b9a56787cc68822a3c1635de82e03ecc0aae27
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@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 b1c3ad79ac..8887bb719a 100644 --- a/src/backends/reference/RefWorkloadFactory.cpp +++ b/src/backends/reference/RefWorkloadFactory.cpp @@ -113,7 +113,11 @@ std::unique_ptr<IWorkload> RefWorkloadFactory::CreateOutput(const OutputQueueDes std::unique_ptr<IWorkload> RefWorkloadFactory::CreateActivation(const ActivationQueueDescriptor& descriptor, const WorkloadInfo& info) const { - return MakeWorkload<RefActivationFloat32Workload, RefActivationUint8Workload>(descriptor, info); + if (IsFloat16(info)) + { + return MakeWorkload<NullWorkload, NullWorkload>(descriptor, info); + } + return std::make_unique<RefActivationWorkload>(descriptor, info); } std::unique_ptr<IWorkload> RefWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& descriptor, |