diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-11-27 16:57:56 +0000 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-12-16 17:38:23 +0000 |
commit | 8719d2247fab4ba0e3936c86e3043da3da1b573c (patch) | |
tree | 70fa1c36560d9484dd1e54f69df2e544394e4d94 /src/armnnTfLiteParser/TfLiteParser.cpp | |
parent | ce55278f775fced323907ada6eb70398ad9e3fa2 (diff) | |
download | armnn-8719d2247fab4ba0e3936c86e3043da3da1b573c.tar.gz |
IVGCVSW-5595 Fix incorrect padding value for asymmetric quantized type
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I85f0c30757043f8c27c78d607f0f9dbbdd35b9fb
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index d1d45f5583..c3d56b13d3 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -1845,6 +1845,8 @@ void TfLiteParser::ParsePad(size_t subgraphIndex, size_t operatorIndex) TfLiteParser::TensorRawPtrVector outputs = GetOutputs(m_Model, subgraphIndex, operatorIndex); CHECK_VALID_SIZE(outputs.size(), 1); + armnn::TensorInfo inputTensorInfo = ToTensorInfo(inputs[0]); + armnn::TensorInfo padTensorInfo = ToTensorInfo(inputs[1]); BufferRawPtr bufferPtr = GetBuffer(m_Model, inputs[1]->buffer); @@ -1853,6 +1855,10 @@ void TfLiteParser::ParsePad(size_t subgraphIndex, size_t operatorIndex) size_t step = 2; armnn::PadDescriptor desc; + if (inputTensorInfo.IsQuantized()) + { + desc.m_PadValue = static_cast<float>(inputTensorInfo.GetQuantizationOffset()); + } for (unsigned int i = 0; i < padTensorInfo.GetNumElements() / step; ++i) { desc.m_PadList.emplace_back(padBuffer[i * step], padBuffer[i * step + 1]); |