diff options
Diffstat (limited to '1.3')
-rw-r--r-- | 1.3/HalPolicy.cpp | 10 | ||||
-rw-r--r-- | 1.3/HalPolicy.hpp | 2 |
2 files changed, 11 insertions, 1 deletions
diff --git a/1.3/HalPolicy.cpp b/1.3/HalPolicy.cpp index b2b8a860..1077b787 100644 --- a/1.3/HalPolicy.cpp +++ b/1.3/HalPolicy.cpp @@ -103,8 +103,10 @@ bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model, return ConvertPrelu(operation, model, data); case V1_3::OperationType::QUANTIZE: return ConvertQuantize(operation, model, data); - case V1_3::OperationType::QUANTIZED_16BIT_LSTM: + case V1_3::OperationType::QUANTIZED_LSTM: return ConvertQuantizedLstm(operation, model, data); + case V1_3::OperationType::QUANTIZED_16BIT_LSTM: + return ConvertQuantized16BitLstm(operation, model, data); case V1_3::OperationType::RELU: return ConvertReLu(operation, model, data); case V1_3::OperationType::RELU1: @@ -370,6 +372,12 @@ bool HalPolicy::ConvertQuantizedLstm(const Operation& operation, const Model& mo return ::ConvertQuantizedLstm<hal_1_3::HalPolicy>(operation, model, data); } +bool HalPolicy::ConvertQuantized16BitLstm(const Operation& operation, const Model& model, ConversionData& data) +{ + ALOGV("hal_1_3::HalPolicy::ConvertQuantized16BitLstm()"); + return ::ConvertQuantized16BitLstm<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 c6019421..b59710a6 100644 --- a/1.3/HalPolicy.hpp +++ b/1.3/HalPolicy.hpp @@ -117,6 +117,8 @@ private: static bool ConvertQuantizedLstm(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertQuantized16BitLstm(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); |