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.3 | |
| 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.3')
| -rw-r--r-- | 1.3/HalPolicy.cpp | 9 | ||||
| -rw-r--r-- | 1.3/HalPolicy.hpp | 2 |
2 files changed, 10 insertions, 1 deletions
diff --git a/1.3/HalPolicy.cpp b/1.3/HalPolicy.cpp index 161ef809..de487423 100644 --- a/1.3/HalPolicy.cpp +++ b/1.3/HalPolicy.cpp @@ -35,6 +35,8 @@ bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model, return ConvertBatchToSpaceNd(operation, model, data); case V1_3::OperationType::CAST: return ConvertCast(operation, model, data); + case V1_3::OperationType::CHANNEL_SHUFFLE: + return ConvertChannelShuffle(operation, model, data); case V1_3::OperationType::CONCATENATION: return ConvertConcatenation(operation, model, data); case V1_3::OperationType::CONV_2D: @@ -208,6 +210,12 @@ bool HalPolicy::ConvertCast(const Operation& operation, const Model& model, Conv return ::ConvertCast<hal_1_3::HalPolicy>(operation, model, data); } +bool HalPolicy::ConvertChannelShuffle(const Operation& operation, const Model& model, ConversionData& data) +{ + ALOGV("hal_1_3::HalPolicy::ConvertChannelShuffle()"); + return ::ConvertChannelShuffle<hal_1_3::HalPolicy>(operation, model, data); +} + bool HalPolicy::ConvertComparison(const Operation& operation, const Model& model, ConversionData& data, @@ -217,7 +225,6 @@ bool HalPolicy::ConvertComparison(const Operation& operation, return ::ConvertComparison_1_2<hal_1_3::HalPolicy>(operation, model, data, comparisonOperation); } - bool HalPolicy::ConvertConcatenation(const Operation& operation, const Model& model, ConversionData& data) { ALOGV("hal_1_3::HalPolicy::ConvertConcatenation()"); diff --git a/1.3/HalPolicy.hpp b/1.3/HalPolicy.hpp index cf1d3665..dee391a7 100644 --- a/1.3/HalPolicy.hpp +++ b/1.3/HalPolicy.hpp @@ -49,6 +49,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, |