diff options
author | David Beck <david.beck@arm.com> | 2018-09-11 15:21:14 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-01 14:56:47 +0100 |
commit | 0a710c4c44be908a93a318e1fbd5c3535e849293 (patch) | |
tree | 3b0f3f1abaa46e404303a5d0b36f14c2af090503 /src/armnn/backends/ClWorkloadFactory.cpp | |
parent | 1952622c4b9f2cecdc93dedf2d6a9a8a94eac312 (diff) | |
download | armnn-0a710c4c44be908a93a318e1fbd5c3535e849293.tar.gz |
IVGCVSW-1843 : refactor ClAdditionWorkload and ClSubtractionWorkload
Change-Id: I0ca9f16217f8e32bb57a49b841611f10dabf021a
Diffstat (limited to 'src/armnn/backends/ClWorkloadFactory.cpp')
-rw-r--r-- | src/armnn/backends/ClWorkloadFactory.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnn/backends/ClWorkloadFactory.cpp b/src/armnn/backends/ClWorkloadFactory.cpp index 056a201783..217c637784 100644 --- a/src/armnn/backends/ClWorkloadFactory.cpp +++ b/src/armnn/backends/ClWorkloadFactory.cpp @@ -154,7 +154,8 @@ std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreateNormalization(const N std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreateAddition(const AdditionQueueDescriptor& descriptor, const WorkloadInfo& info) const { - return MakeWorkload<ClAdditionFloatWorkload, ClAdditionUint8Workload>(descriptor, info); + return MakeWorkload<ClAdditionWorkload<armnn::DataType::Float16, armnn::DataType::Float32>, + ClAdditionWorkload<armnn::DataType::QuantisedAsymm8>>(descriptor, info); } std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreateMultiplication( @@ -172,7 +173,8 @@ std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreateDivision( std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreateSubtraction(const SubtractionQueueDescriptor& descriptor, const WorkloadInfo& info) const { - return MakeWorkload<ClSubtractionFloatWorkload, ClSubtractionUint8Workload>(descriptor, info); + return MakeWorkload<ClSubtractionWorkload<armnn::DataType::Float16, armnn::DataType::Float32>, + ClSubtractionWorkload<armnn::DataType::QuantisedAsymm8>>(descriptor, info); } std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreateBatchNormalization( |