diff options
author | Cathal Corbett <cathal.corbett@arm.com> | 2022-12-09 12:17:27 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2022-12-13 12:13:20 +0000 |
commit | 3b9acd515918ac1af5498b3e7501c0b16a88a8e0 (patch) | |
tree | a537cbdd7c45fea1772cbb34f8a23ecc7dbd6e1a /src/backends/tosaCommon/TosaMappings.cpp | |
parent | 8107ccce5167f160b9c98a6042878dd0408220b5 (diff) | |
download | armnn-3b9acd515918ac1af5498b3e7501c0b16a88a8e0.tar.gz |
IVGCVSW-7342 Add Slice support to TOSA Reference Backend
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: I8be286b69bebd4cd36033e3145632bb043938d16
Diffstat (limited to 'src/backends/tosaCommon/TosaMappings.cpp')
-rw-r--r-- | src/backends/tosaCommon/TosaMappings.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/tosaCommon/TosaMappings.cpp b/src/backends/tosaCommon/TosaMappings.cpp index 318735db77..15629ffab0 100644 --- a/src/backends/tosaCommon/TosaMappings.cpp +++ b/src/backends/tosaCommon/TosaMappings.cpp @@ -62,6 +62,11 @@ TosaSerializationBasicBlock* GetTosaMapping(const Layer* layer, auto reshapeDesc = PolymorphicDowncast<const ReshapeDescriptor*>(&descriptor); return ConvertReshapeToTosaOperator(layer, inputs, outputs, reshapeDesc); } + case LayerType::Slice: + { + auto sliceDesc = PolymorphicDowncast<const SliceDescriptor*>(&descriptor); + return ConvertSliceToTosaOperator(layer, inputs, outputs, sliceDesc); + } default: { return CreateEmptyTosaSerializationBasicBlock(); |