aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonWorkloadFactory.cpp
diff options
context:
space:
mode:
authorTianle Cheng <tianle.cheng@arm.com>2023-11-09 13:56:53 +0000
committerTianle Cheng <tianle.cheng@arm.com>2023-11-10 17:39:25 +0000
commit21a9f33338c60ae1cd955df220ce329918adcb8f (patch)
tree0c684061fbfa88257bafa532eeab74cc06365350 /src/backends/neon/NeonWorkloadFactory.cpp
parent37acf49013ece8a3490174a4bfc1803e6f23b956 (diff)
downloadarmnn-21a9f33338c60ae1cd955df220ce329918adcb8f.tar.gz
IVGCVSW-7835 Add ReverseV2 CL and Neon Workloads
* Added ReverseV2 to CL and Neon backends * Added Cl and Neon ReverseV2 Layer unit tests Signed-off-by: Tianle Cheng <tianle.cheng@arm.com> Change-Id: I646275c629caf17dac1950b0cd7083f23f87f387
Diffstat (limited to 'src/backends/neon/NeonWorkloadFactory.cpp')
-rw-r--r--src/backends/neon/NeonWorkloadFactory.cpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/neon/NeonWorkloadFactory.cpp b/src/backends/neon/NeonWorkloadFactory.cpp
index 4f131ac575..9b74f750f9 100644
--- a/src/backends/neon/NeonWorkloadFactory.cpp
+++ b/src/backends/neon/NeonWorkloadFactory.cpp
@@ -572,6 +572,11 @@ std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateWorkload(LayerType type,
auto resizeQueueDescriptor = PolymorphicDowncast<const ResizeQueueDescriptor*>(&descriptor);
return std::make_unique<NeonResizeWorkload>(*resizeQueueDescriptor, info);
}
+ case LayerType::ReverseV2 :
+ {
+ auto reverseV2QueueDescriptor = PolymorphicDowncast<const ReverseV2QueueDescriptor*>(&descriptor);
+ return std::make_unique<NeonReverseV2Workload>(*reverseV2QueueDescriptor, info);
+ }
case LayerType::Slice :
{
auto sliceQueueDescriptor = PolymorphicDowncast<const SliceQueueDescriptor*>(&descriptor);