diff options
author | Ruomei Yan <ruomei.yan@arm.com> | 2019-05-23 11:37:33 +0100 |
---|---|---|
committer | Ruomei Yan <ruomei.yan@arm.com> | 2019-05-23 11:51:39 +0100 |
commit | 495852f2adef1d11fbf13ce6347cf61973ce1a65 (patch) | |
tree | 9c26ac47362548e001ad7e5203212d2ef3226e9d /src/backends/reference/RefWorkloadFactory.cpp | |
parent | 9b3983299f882c8d84c5abd0d40ca75a801ea7f2 (diff) | |
download | armnn-495852f2adef1d11fbf13ce6347cf61973ce1a65.tar.gz |
IVGCVSW-3073 Refactor reference DepthwiseConvolution2d workloads
Change-Id: I3cf8d9dbc4f8c95e0e2311505dd7e9f9069f1ab5
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 | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/src/backends/reference/RefWorkloadFactory.cpp b/src/backends/reference/RefWorkloadFactory.cpp index be920943d1..2266fcd2f2 100644 --- a/src/backends/reference/RefWorkloadFactory.cpp +++ b/src/backends/reference/RefWorkloadFactory.cpp @@ -163,11 +163,10 @@ std::unique_ptr<armnn::IWorkload> RefWorkloadFactory::CreateConvolution2d( return std::make_unique<RefConvolution2dWorkload>(descriptor, info); } -std::unique_ptr<IWorkload> RefWorkloadFactory::CreateDepthwiseConvolution2d( +std::unique_ptr<armnn::IWorkload> RefWorkloadFactory::CreateDepthwiseConvolution2d( const DepthwiseConvolution2dQueueDescriptor& descriptor, const WorkloadInfo& info) const { - return MakeWorkload<RefDepthwiseConvolution2dFloat32Workload, - RefDepthwiseConvolution2dUint8Workload>(descriptor, info); + return std::make_unique<RefDepthwiseConvolution2dWorkload>(descriptor, info); } std::unique_ptr<IWorkload> RefWorkloadFactory::CreateDetectionPostProcess( |