diff options
| author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-10-15 13:33:03 +0100 |
|---|---|---|
| committer | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-10-15 13:33:03 +0100 |
| commit | 75e677939a98298e50d65c4a7e99a03fb51d5e3c (patch) | |
| tree | 69a32790256c26a94a45f31641179a37373f0bc8 /1.2/HalPolicy.hpp | |
| parent | ad7545385e42bd183f6d27bfea30a57f76a1fb46 (diff) | |
| download | android-nn-driver-75e677939a98298e50d65c4a7e99a03fb51d5e3c.tar.gz | |
IVGCVSW-3894 Add support for LOG_SOFTMAX to the HAL 1.2 driver
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I59645b339f3b176e5d0852769acb95f5657101d3
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 aa69f127..743ac11e 100644 --- a/1.2/HalPolicy.hpp +++ b/1.2/HalPolicy.hpp @@ -71,6 +71,8 @@ private: static bool ConvertLogistic(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertLogSoftmax(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertLstm(const Operation& operation, const Model& model, ConversionData& data); static bool ConvertMaxPool2d(const Operation& operation, const Model& model, ConversionData& data); |