From 4e1e136cce3fca73ba49b570cfcb620f4ec574da Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=89anna=20=C3=93=20Cath=C3=A1in?= Date: Mon, 12 Nov 2018 11:36:34 +0000 Subject: IVGCVSW-2054: BATCH_TO_SPACE_ND Reference implementation and Unit tests. Change-Id: I13c6728dbb60643d0e086d171225c5d802987f92 --- src/backends/neon/NeonWorkloadFactory.cpp | 12 ++++++++++++ src/backends/neon/NeonWorkloadFactory.hpp | 3 +++ 2 files changed, 15 insertions(+) (limited to 'src/backends/neon') diff --git a/src/backends/neon/NeonWorkloadFactory.cpp b/src/backends/neon/NeonWorkloadFactory.cpp index 11b5634a79..f0d916b63b 100644 --- a/src/backends/neon/NeonWorkloadFactory.cpp +++ b/src/backends/neon/NeonWorkloadFactory.cpp @@ -279,6 +279,12 @@ std::unique_ptr NeonWorkloadFactory::CreatePad(const PadQueueDescript return MakeWorkloadHelper(descriptor, info); } +std::unique_ptr NeonWorkloadFactory::CreateBatchToSpaceNd(const BatchToSpaceNdQueueDescriptor& descriptor, + const WorkloadInfo& info) const +{ + return MakeWorkloadHelper(descriptor, info); +} + void NeonWorkloadFactory::Release() { m_MemoryManager.Release(); @@ -495,6 +501,12 @@ std::unique_ptr NeonWorkloadFactory::CreatePad(const PadQueueDescript return nullptr; } +std::unique_ptr NeonWorkloadFactory::CreateBatchToSpaceNd(const BatchToSpaceNdQueueDescriptor& descriptor, + const WorkloadInfo& info) const +{ + return nullptr; +} + void NeonWorkloadFactory::Release() {} diff --git a/src/backends/neon/NeonWorkloadFactory.hpp b/src/backends/neon/NeonWorkloadFactory.hpp index 9b574f6a14..98f323afdf 100644 --- a/src/backends/neon/NeonWorkloadFactory.hpp +++ b/src/backends/neon/NeonWorkloadFactory.hpp @@ -128,6 +128,9 @@ public: virtual std::unique_ptr CreatePad(const PadQueueDescriptor& descriptor, const WorkloadInfo& info) const override; + virtual std::unique_ptr CreateBatchToSpaceNd(const BatchToSpaceNdQueueDescriptor& descriptor, + const WorkloadInfo& Info) const override; + virtual void Release() override; virtual void Acquire() override; -- cgit v1.2.1