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/HalPolicy.cpp | |
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/HalPolicy.cpp')
-rw-r--r-- | 1.3/HalPolicy.cpp | 8 |
1 files changed, 8 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()"); |