diff options
| author | Francis Murtagh <francis.murtagh@arm.com> | 2019-07-22 16:40:57 +0100 |
|---|---|---|
| committer | Francis Murtagh <francis.murtagh@arm.com> | 2019-07-22 16:40:57 +0100 |
| commit | 074c25a1535b648fdf19d7f6648e8aceef9aa7ad (patch) | |
| tree | 87fa092745acac3f7788cb7412078105b5c7e6b1 /1.2/HalPolicy.hpp | |
| parent | 65c42dc4d68ac163b77a3139feee3e7d4530b5c5 (diff) | |
| download | android-nn-driver-074c25a1535b648fdf19d7f6648e8aceef9aa7ad.tar.gz | |
IVGCVSW-3140 Fix Hal 1.1 Softmax failures on Android Q
* Add support for Softmax Axis parameter.
!armnn:1567
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Change-Id: I7e47d36f13b122dbad7976c0d59773845bc182b1
Diffstat (limited to '1.2/HalPolicy.hpp')
| -rw-r--r-- | 1.2/HalPolicy.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/1.2/HalPolicy.hpp b/1.2/HalPolicy.hpp index 18cf0359..3c4906cb 100644 --- a/1.2/HalPolicy.hpp +++ b/1.2/HalPolicy.hpp @@ -48,6 +48,8 @@ private: ConversionData& data, armnn::ResizeMethod resizeMethod); + static bool ConvertSoftmax(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertSpaceToDepth(const Operation& operation, const Model& model, ConversionData& data); }; |