diff options
author | Tianle Cheng <tianle.cheng@arm.com> | 2023-11-09 13:56:53 +0000 |
---|---|---|
committer | Tianle Cheng <tianle.cheng@arm.com> | 2023-11-10 17:39:25 +0000 |
commit | 21a9f33338c60ae1cd955df220ce329918adcb8f (patch) | |
tree | 0c684061fbfa88257bafa532eeab74cc06365350 /src/backends/neon/NeonWorkloadFactory.cpp | |
parent | 37acf49013ece8a3490174a4bfc1803e6f23b956 (diff) | |
download | armnn-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.cpp | 5 |
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); |