diff options
Diffstat (limited to '1.0')
| -rw-r--r-- | 1.0/HalPolicy.cpp | 8 | ||||
| -rw-r--r-- | 1.0/HalPolicy.hpp | 2 |
2 files changed, 10 insertions, 0 deletions
diff --git a/1.0/HalPolicy.cpp b/1.0/HalPolicy.cpp index cff678a8..7e9e9efa 100644 --- a/1.0/HalPolicy.cpp +++ b/1.0/HalPolicy.cpp @@ -27,6 +27,8 @@ bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model, return ConvertConcatenation(operation, model, data); case V1_0::OperationType::CONV_2D: return ConvertConv2d(operation, model, data); + case V1_0::OperationType::DEPTH_TO_SPACE: + return ConvertDepthToSpace(operation, model, data); case V1_0::OperationType::DEPTHWISE_CONV_2D: return ConvertDepthwiseConv2d(operation, model, data); case V1_0::OperationType::DEQUANTIZE: @@ -95,6 +97,12 @@ bool HalPolicy::ConvertConv2d(const Operation& operation, const Model& model, Co return ::ConvertConv2d<hal_1_0::HalPolicy>(operation, model, data); } +bool HalPolicy::ConvertDepthToSpace(const Operation& operation, const Model& model, ConversionData& data) +{ + ALOGV("hal_1_0::HalPolicy::ConvertDepthToSpace()"); + return ::ConvertDepthToSpace<hal_1_0::HalPolicy>(operation, model, data); +} + bool HalPolicy::ConvertDepthwiseConv2d(const Operation& operation, const Model& model, ConversionData& data) { ALOGV("hal_1_0::HalPolicy::ConvertDepthwiseConv2d()"); diff --git a/1.0/HalPolicy.hpp b/1.0/HalPolicy.hpp index 4b8dc47a..9eb13b47 100644 --- a/1.0/HalPolicy.hpp +++ b/1.0/HalPolicy.hpp @@ -38,6 +38,8 @@ private: static bool ConvertConv2d(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertDepthToSpace(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertDepthwiseConv2d(const Operation& operation, const Model& model, ConversionData& data); static bool ConvertDequantize(const Operation& operation, const Model& model, ConversionData& data); |