diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2021-04-26 09:52:06 +0100 |
|---|---|---|
| committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-04-26 09:52:06 +0100 |
| commit | 92b5fd1b1f16fafa59d1757fe2c464e60181a55e (patch) | |
| tree | 659ed7400fd873b581f10917ce8522adda8f0397 /1.2 | |
| parent | a71c0633eff4791eb98362f72f198a2d1ec3d8f9 (diff) | |
| download | android-nn-driver-92b5fd1b1f16fafa59d1757fe2c464e60181a55e.tar.gz | |
IVGCVSW-5416 'Add android-nn-driver support for CAST'
!armnn:5494
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Ie99d58f7421c2634c23d2a68d1acaaac651963a4
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 e96c4cb0..77511311 100644 --- a/1.2/HalPolicy.cpp +++ b/1.2/HalPolicy.cpp @@ -33,6 +33,8 @@ bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model, return ConvertAveragePool2d(operation, model, data); case V1_2::OperationType::BATCH_TO_SPACE_ND: return ConvertBatchToSpaceNd(operation, model, data); + case V1_2::OperationType::CAST: + return ConvertCast(operation, model, data); case V1_2::OperationType::CONCATENATION: return ConvertConcatenation(operation, model, data); case V1_2::OperationType::CONV_2D: @@ -178,6 +180,12 @@ bool HalPolicy::ConvertBatchToSpaceNd(const Operation& operation, const Model& m return ::ConvertBatchToSpaceNd<hal_1_2::HalPolicy>(operation, model, data); } +bool HalPolicy::ConvertCast(const Operation& operation, const Model& model, ConversionData& data) +{ + ALOGV("hal_1_2::HalPolicy::ConvertCast()"); + return ::ConvertCast<hal_1_2::HalPolicy>(operation, model, data); +} + bool HalPolicy::ConvertComparison(const Operation& operation, const Model& model, ConversionData& data, diff --git a/1.2/HalPolicy.hpp b/1.2/HalPolicy.hpp index abd60e72..a348abe0 100644 --- a/1.2/HalPolicy.hpp +++ b/1.2/HalPolicy.hpp @@ -46,6 +46,8 @@ private: static bool ConvertBatchToSpaceNd(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertCast(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertComparison(const Operation& operation, const Model& model, ConversionData& data, |