diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2019-11-19 12:06:47 +0000 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2019-11-19 12:06:55 +0000 |
commit | 19fa0cc4fd4915451c3f56f6f4bdb51d8a5ac172 (patch) | |
tree | 62383c52829cac3f53d6aeab69715061062abb6e /1.2/HalPolicy.hpp | |
parent | 0e2e31b9695f58181e15e770fcfe3d92aa09bf47 (diff) | |
download | android-nn-driver-19fa0cc4fd4915451c3f56f6f4bdb51d8a5ac172.tar.gz |
IVGCVSW-3697 Add Support for ANEURALNETWORKS_ARG[MAX|MIN] to HAL 1.2 Driver
!armnn:2313
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Change-Id: I543136e4e2ef9aece1378d2642064cc585246645
Diffstat (limited to '1.2/HalPolicy.hpp')
-rw-r--r-- | 1.2/HalPolicy.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/1.2/HalPolicy.hpp b/1.2/HalPolicy.hpp index d611102b..1f0b54d7 100644 --- a/1.2/HalPolicy.hpp +++ b/1.2/HalPolicy.hpp @@ -35,6 +35,11 @@ private: static bool ConvertAdd(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertArgMinMax(const Operation& operation, + const Model& model, + ConversionData& data, + armnn::ArgMinMaxFunction argMinMaxFunction); + static bool ConvertAveragePool2d(const Operation& operation, const Model& model, ConversionData& data); static bool ConvertBatchToSpaceNd(const Operation& operation, const Model& model, ConversionData& data); |