From 89cbb3a914d76843630f3576108e309cf085ef80 Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Thu, 11 Feb 2021 21:00:47 +0000 Subject: IVGCVSW-5131 Add support for REDUCE_SUM, REDUCE_MIN and REDUCE_MAX Signed-off-by: Teresa Charlin Change-Id: Ib88544dd6adc3d6a0d2f4c0b395e05b89bc8ad3e --- 1.3/HalPolicy.cpp | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) (limited to '1.3/HalPolicy.cpp') diff --git a/1.3/HalPolicy.cpp b/1.3/HalPolicy.cpp index 93337594..9a9f2ac0 100644 --- a/1.3/HalPolicy.cpp +++ b/1.3/HalPolicy.cpp @@ -98,7 +98,7 @@ bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model, case V1_3::OperationType::MAXIMUM: return ConvertMaximum(operation, model, data); case V1_3::OperationType::MEAN: - return ConvertMean(operation, model, data); + return ConvertReduce(operation, model, data, ReduceOperation::Mean); case V1_3::OperationType::MINIMUM: return ConvertMinimum(operation, model, data); case V1_3::OperationType::MUL: @@ -121,6 +121,12 @@ bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model, return ConvertQuantized16BitLstm(operation, model, data); case V1_3::OperationType::RANK: return ConvertRank(operation, model, data); + case V1_3::OperationType::REDUCE_MAX: + return ConvertReduce(operation, model, data, ReduceOperation::Max); + case V1_3::OperationType::REDUCE_MIN: + return ConvertReduce(operation, model, data, ReduceOperation::Min); + case V1_3::OperationType::REDUCE_SUM: + return ConvertReduce(operation, model, data, ReduceOperation::Sum); case V1_3::OperationType::RELU: return ConvertReLu(operation, model, data); case V1_3::OperationType::RELU1: @@ -359,10 +365,11 @@ bool HalPolicy::ConvertMaximum(const Operation& operation, const Model& model, C return ::ConvertMaximum(operation, model, data); } -bool HalPolicy::ConvertMean(const Operation& operation, const Model& model, ConversionData& data) +bool HalPolicy::ConvertReduce(const Operation& operation, const Model& model, ConversionData& data, + ReduceOperation reduceOperation) { - ALOGV("hal_1_3::HalPolicy::ConvertMean()"); - return ::ConvertMean(operation, model, data); + ALOGV("hal_1_3::HalPolicy::ConvertReduce()"); + return ::ConvertReduce(operation, model, data, reduceOperation); } bool HalPolicy::ConvertMinimum(const Operation& operation, const Model& model, ConversionData& data) -- cgit v1.2.1