diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-09-01 14:19:38 +0100 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-09-23 14:08:03 +0100 |
| commit | 7f5b51e6b66771bdb17be056ddee386fd8a79e3e (patch) | |
| tree | 50a3d8befad9f17100fce004c238755a8367e329 /1.2 | |
| parent | 6bf3f43c6f1d88c676f277f4c3c766cb3ea37ed0 (diff) | |
| download | android-nn-driver-7f5b51e6b66771bdb17be056ddee386fd8a79e3e.tar.gz | |
IVGCVSW-3715 Add Channel Shuffle support
!armnn:6211
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ie5e5d160f6d826c30f1fff298a4a73151c1ccdbb
Diffstat (limited to '1.2')
| -rw-r--r-- | 1.2/HalPolicy.cpp | 8 | ||||
| -rw-r--r-- | 1.2/HalPolicy.hpp | 2 |
2 files changed, 10 insertions, 0 deletions
diff --git a/1.2/HalPolicy.cpp b/1.2/HalPolicy.cpp index 79d117ae..4cefd599 100644 --- a/1.2/HalPolicy.cpp +++ b/1.2/HalPolicy.cpp @@ -64,6 +64,8 @@ bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model, return ConvertBatchToSpaceNd(operation, model, data); case V1_2::OperationType::CAST: return ConvertCast(operation, model, data); + case V1_2::OperationType::CHANNEL_SHUFFLE: + return ConvertChannelShuffle(operation, model, data); case V1_2::OperationType::CONCATENATION: return ConvertConcatenation(operation, model, data); case V1_2::OperationType::CONV_2D: @@ -221,6 +223,12 @@ bool HalPolicy::ConvertCast(const Operation& operation, const Model& model, Conv return ::ConvertCast<hal_1_2::HalPolicy>(operation, model, data); } +bool HalPolicy::ConvertChannelShuffle(const Operation& operation, const Model& model, ConversionData& data) +{ + ALOGV("hal_1_2::HalPolicy::ConvertChannelShuffle()"); + return ::ConvertChannelShuffle<hal_1_2::HalPolicy>(operation, model, data); +} + bool HalPolicy::ConvertComparison(const Operation& operation, const Model& model, ConversionData& data, diff --git a/1.2/HalPolicy.hpp b/1.2/HalPolicy.hpp index 0662e1be..9fb74579 100644 --- a/1.2/HalPolicy.hpp +++ b/1.2/HalPolicy.hpp @@ -52,6 +52,8 @@ private: static bool ConvertCast(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertChannelShuffle(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertComparison(const Operation& operation, const Model& model, ConversionData& data, |