diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-03-12 09:12:29 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-03-12 09:12:29 +0000 |
commit | 6a2845317f9f9e5161f3afab2b54dac494f57f89 (patch) | |
tree | fbaaae4f162b80adc5998d2b77641a80d0bed59e /1.2/HalPolicy.hpp | |
parent | 89cbb3a914d76843630f3576108e309cf085ef80 (diff) | |
download | android-nn-driver-6a2845317f9f9e5161f3afab2b54dac494f57f89.tar.gz |
IVGCVSW-5767 'Regression on SqueezeNet v1.1'
* MEAN operator second input data type is different than Reduction operators
* Use existing ConvertMean() function for MEAN operator
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I211cb77515d88e54ded9646eebe3e901d437827c
Diffstat (limited to '1.2/HalPolicy.hpp')
-rw-r--r-- | 1.2/HalPolicy.hpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/1.2/HalPolicy.hpp b/1.2/HalPolicy.hpp index feb2ba20..abd60e72 100644 --- a/1.2/HalPolicy.hpp +++ b/1.2/HalPolicy.hpp @@ -98,8 +98,7 @@ private: static bool ConvertMaximum(const Operation& operation, const Model& model, ConversionData& data); - static bool ConvertReduce(const Operation& operation, const Model& model, ConversionData& data, - ReduceOperation reduce_operation); + static bool ConvertMean(const Operation& operation, const Model& model, ConversionData& data); static bool ConvertMinimum(const Operation& operation, const Model& model, ConversionData& data); @@ -115,6 +114,11 @@ private: static bool ConvertQuantized16BitLstm(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertReduce(const Operation& operation, + const Model& model, + ConversionData& data, + ReduceOperation reduce_operation); + static bool ConvertReLu(const Operation& operation, const Model& model, ConversionData& data); static bool ConvertReLu1(const Operation& operation, const Model& model, ConversionData& data); |