diff options
| author | Finn Williams <Finn.Williams@Arm.com> | 2020-06-11 17:35:44 +0100 |
|---|---|---|
| committer | Finn Williams <Finn.Williams@Arm.com> | 2020-07-06 19:07:52 +0100 |
| commit | fc884b4141a28fbd3c62f665341ec88158fcd332 (patch) | |
| tree | df18555baced3f4d917a78d24d87c5e2d1778f83 /1.3 | |
| parent | 00b586b1f3106a4e6970ca7feacb1cc1892d107e (diff) | |
| download | android-nn-driver-fc884b4141a28fbd3c62f665341ec88158fcd332.tar.gz | |
Add support for Rank and scalar tensors
!armnn:3330
Signed-off-by: Finn Williams <Finn.Williams@Arm.com>
Change-Id: Icc429d9fabb570193d12bffef0e00dda7b51032f
Diffstat (limited to '1.3')
| -rw-r--r-- | 1.3/HalPolicy.cpp | 8 | ||||
| -rw-r--r-- | 1.3/HalPolicy.hpp | 2 |
2 files changed, 10 insertions, 0 deletions
diff --git a/1.3/HalPolicy.cpp b/1.3/HalPolicy.cpp index 1c4a1e36..79df1c7f 100644 --- a/1.3/HalPolicy.cpp +++ b/1.3/HalPolicy.cpp @@ -111,6 +111,8 @@ bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model, return ConvertQuantizedLstm(operation, model, data); case V1_3::OperationType::QUANTIZED_16BIT_LSTM: return ConvertQuantized16BitLstm(operation, model, data); + case V1_3::OperationType::RANK: + return ConvertRank(operation, model, data); case V1_3::OperationType::RELU: return ConvertReLu(operation, model, data); case V1_3::OperationType::RELU1: @@ -394,6 +396,12 @@ bool HalPolicy::ConvertQuantized16BitLstm(const Operation& operation, const Mode return ::ConvertQuantized16BitLstm<hal_1_3::HalPolicy>(operation, model, data); } +bool HalPolicy::ConvertRank(const Operation& operation, const Model& model, ConversionData& data) +{ + ALOGV("hal_1_3::HalPolicy::ConvertRank()"); + return ::ConvertRank<hal_1_3::HalPolicy>(operation, model, data); +} + bool HalPolicy::ConvertReLu(const Operation& operation, const Model& model, ConversionData& data) { ALOGV("hal_1_3::HalPolicy::ConvertReLu()"); diff --git a/1.3/HalPolicy.hpp b/1.3/HalPolicy.hpp index 6df2ce2d..0eb5f4d7 100644 --- a/1.3/HalPolicy.hpp +++ b/1.3/HalPolicy.hpp @@ -123,6 +123,8 @@ private: static bool ConvertQuantized16BitLstm(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertRank(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertReLu(const Operation& operation, const Model& model, ConversionData& data); static bool ConvertReLu1(const Operation& operation, const Model& model, ConversionData& data); |