diff options
| author | Cathal Corbett <cathal.corbett@arm.com> | 2022-01-21 16:55:13 +0000 |
|---|---|---|
| committer | Cathal Corbett <cathal.corbett@arm.com> | 2022-01-24 18:10:44 +0000 |
| commit | 0fa5e6dda7e9e76d7ca5f77777c35374c768f28c (patch) | |
| tree | fe4cdfe345398429328be5e5485cea5d3a0073be /1.3/HalPolicy.hpp | |
| parent | e02562b0957b4c14739f608513d3c03050c19369 (diff) | |
| download | android-nn-driver-0fa5e6dda7e9e76d7ca5f77777c35374c768f28c.tar.gz | |
IVGCVSW-6269 Add support of Unidirectional Sequence Lstm fp32/fp16 to Android
* Unidirectional Sequence Lstm is supported in android-nn-driver.
* CTS and VTS pass for Unidirectional Sequence Lstm if there are
any without weights as inputs. If that isn't the case use the
tests implemented in the android driver for LSTM.
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: Ie3a237b67f5101c4f9a2c7bea796e9c674bedba1
Diffstat (limited to '1.3/HalPolicy.hpp')
| -rw-r--r-- | 1.3/HalPolicy.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/1.3/HalPolicy.hpp b/1.3/HalPolicy.hpp index dee391a7..7411b24b 100644 --- a/1.3/HalPolicy.hpp +++ b/1.3/HalPolicy.hpp @@ -171,6 +171,10 @@ private: static bool ConvertTranspose(const Operation& operation, const Model& model, ConversionData& data); static bool ConvertTransposeConv2d(const Operation& operation, const Model& model, ConversionData& data); + + static bool ConvertUnidirectionalSequenceLstm(const Operation& operation, + const Model& model, + ConversionData& data); }; } // namespace hal_1_3 |