diff options
author | Cathal Corbett <cathal.corbett@arm.com> | 2022-04-15 14:12:08 +0100 |
---|---|---|
committer | Cathal Corbett <cathal.corbett@arm.com> | 2022-05-05 09:11:47 +0000 |
commit | 915f2a7327098636cc5a292510cd00b715e67e90 (patch) | |
tree | 20933b85c8e2f46f97c82031b7925c176d4ccf7b /ConversionUtils.cpp | |
parent | 0ae102a0fc9d66b7067cf8d7a0ed1af5ed65ae50 (diff) | |
download | android-nn-driver-915f2a7327098636cc5a292510cd00b715e67e90.tar.gz |
IVGCVSW-6127 ConstTensorsAsInput: DepthwiseConvolution2d
!armnn:7417
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: Ic37d6b8677e040c60a90358dd0a4a8eb33fb6ea0
Diffstat (limited to 'ConversionUtils.cpp')
-rw-r--r-- | ConversionUtils.cpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/ConversionUtils.cpp b/ConversionUtils.cpp index 7eb7343e..91c1ae4e 100644 --- a/ConversionUtils.cpp +++ b/ConversionUtils.cpp @@ -59,6 +59,19 @@ const armnn::TensorInfo& LayerInputHandle::GetTensorInfo() const return m_TensorInfo; } +void LayerInputHandle::SanitizeQuantizationScale(LayerInputHandle& weight, + LayerInputHandle& input) +{ + armnn::TensorInfo weightInfo = weight.GetTensorInfo(); + armnn::TensorInfo inputInfo = input.GetTensorInfo(); + armnn::TensorInfo biasInfo = GetTensorInfo(); + + SanitizeBiasQuantizationScale(biasInfo, weightInfo, inputInfo); + + m_TensorInfo = biasInfo; + m_OutputSlot->SetTensorInfo(biasInfo); +} + ConstTensorPin::ConstTensorPin(bool optional) : m_Optional(optional) {} |