diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2020-04-10 16:46:53 +0100 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-05-29 13:55:02 +0000 |
| commit | f931af987c63466c95426742d7297d49438f8170 (patch) | |
| tree | 8382aac349083d0322229de98ab37ea21dca6d71 /1.2 | |
| parent | e75d09ed83d15bd46f92068c2ba8e293bf02a010 (diff) | |
| download | android-nn-driver-f931af987c63466c95426742d7297d49438f8170.tar.gz | |
IVGCVSW-3847 Add Support for GATHER
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I69dd78d47628355c207a450119b054b04581c729
Diffstat (limited to '1.2')
| -rw-r--r-- | 1.2/HalPolicy.cpp | 8 | ||||
| -rw-r--r-- | 1.2/HalPolicy.hpp | 2 |
2 files changed, 10 insertions, 0 deletions
diff --git a/1.2/HalPolicy.cpp b/1.2/HalPolicy.cpp index d55e587f..4c2a6b5d 100644 --- a/1.2/HalPolicy.cpp +++ b/1.2/HalPolicy.cpp @@ -53,6 +53,8 @@ bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model, return ConvertFloor(operation, model, data); case V1_2::OperationType::FULLY_CONNECTED: return ConvertFullyConnected(operation, model, data); + case V1_2::OperationType::GATHER: + return ConvertGather(operation, model, data); case V1_2::OperationType::GREATER: return ConvertComparison(operation, model, data, ComparisonOperation::Greater); case V1_2::OperationType::GREATER_EQUAL: @@ -240,6 +242,12 @@ bool HalPolicy::ConvertFullyConnected(const Operation& operation, const Model& m return ::ConvertFullyConnected<hal_1_2::HalPolicy>(operation, model, data); } +bool HalPolicy::ConvertGather (const Operation& operation, const Model& model, ConversionData& data) +{ + ALOGV("hal_1_2::HalPolicy::ConvertGather()"); + return ::ConvertGather<hal_1_2::HalPolicy>(operation, model, data); +} + bool HalPolicy::ConvertGroupedConv2d(const Operation& operation, const Model& model, ConversionData& data) { ALOGV("hal_1_2::HalPolicy::ConvertGroupedConv2d()"); diff --git a/1.2/HalPolicy.hpp b/1.2/HalPolicy.hpp index ce43a6e5..be02c22f 100644 --- a/1.2/HalPolicy.hpp +++ b/1.2/HalPolicy.hpp @@ -74,6 +74,8 @@ private: static bool ConvertFullyConnected(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertGather(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertGroupedConv2d(const Operation& operation, const Model& model, ConversionData& data); static bool ConvertInstanceNormalization(const Operation& operation, const Model& model, ConversionData& data); |