diff options
| author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-06-12 15:06:53 +0100 |
|---|---|---|
| committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-06-12 14:31:26 +0000 |
| commit | e205318c65191990ad5f3e388a0301bc91d88696 (patch) | |
| tree | 661f8673cc1c57dc8208b08fb56d324b1d09b4c5 /ConversionUtils.hpp | |
| parent | 07c7c9a38dbd08bb2f2573e5cfc6bbd9090020dc (diff) | |
| download | android-nn-driver-e205318c65191990ad5f3e388a0301bc91d88696.tar.gz | |
IVGCVSW-3265 Add support for optional Convolution2d dilation parameters
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I3968d682b67439d4b4aca670d074ea4f8993a39a
Diffstat (limited to 'ConversionUtils.hpp')
| -rw-r--r-- | ConversionUtils.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp index 808fc1ca..2f26c120 100644 --- a/ConversionUtils.hpp +++ b/ConversionUtils.hpp @@ -1181,7 +1181,8 @@ bool ConvertConv2d(const HalOperation& operation, const HalModel& model, Convers data) || !GetInputScalar<HalOperand, HalOperandType>(operation, 8, HalOperandType::INT32, desc.m_StrideY, model, data) - || !GetInputActivationFunction<HalOperand, HalOperandType>(operation, 9, activation, model, data)) + || !GetInputActivationFunction<HalOperand, HalOperandType>(operation, 9, activation, model, data) + || !GetOptionalConvolutionDilationParams<HalOperand, HalOperandType>(operation, 11, desc, model, data)) { return Fail("%s: Operation has invalid inputs", __func__); } @@ -1195,7 +1196,8 @@ bool ConvertConv2d(const HalOperation& operation, const HalModel& model, Convers model, data) || !GetInputScalar<HalOperand, HalOperandType>(operation, 5, HalOperandType::INT32, desc.m_StrideY, model, data) - || !GetInputActivationFunction<HalOperand, HalOperandType>(operation, 6, activation, model, data)) + || !GetInputActivationFunction<HalOperand, HalOperandType>(operation, 6, activation, model, data) + || !GetOptionalConvolutionDilationParams<HalOperand, HalOperandType>(operation, 8, desc, model, data)) { return Fail("%s: Operation has invalid inputs", __func__); } |