diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-29 15:25:04 +0000 |
---|---|---|
committer | nattapat.chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-29 16:37:44 +0000 |
commit | 8a54ac09dc4b9be1c1826bc63d4baf5101dafd53 (patch) | |
tree | 4bd29824f03b734298ccff7d86a581914275b967 /src/backends/reference/workloads/RefDequantizeWorkload.cpp | |
parent | 317cae526ba7b58891b6f054bb10af3b6e22d5a3 (diff) | |
download | armnn-8a54ac09dc4b9be1c1826bc63d4baf5101dafd53.tar.gz |
IVGCVSW-2875 Reference implementation and unit tests for Dequantize
Change-Id: Ie4ade0519cb0bbe35dc36be6c9cd749b9171c74b
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'src/backends/reference/workloads/RefDequantizeWorkload.cpp')
-rw-r--r-- | src/backends/reference/workloads/RefDequantizeWorkload.cpp | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/RefDequantizeWorkload.cpp b/src/backends/reference/workloads/RefDequantizeWorkload.cpp new file mode 100644 index 0000000000..d861c50730 --- /dev/null +++ b/src/backends/reference/workloads/RefDequantizeWorkload.cpp @@ -0,0 +1,34 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include "RefDequantizeWorkload.hpp" +#include "RefWorkloadUtils.hpp" + +namespace armnn +{ + +void RefDequantizeWorkload::Execute() const +{ + ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefDequantizeWorkload_Execute"); + + const TensorInfo& inputInfo = GetTensorInfo(m_Data.m_Inputs[0]); + const DataType& inputDataType = inputInfo.GetDataType(); + + float* outputData = GetOutputTensorData<float>(0, m_Data); + + switch (inputDataType) + { + case DataType::QuantisedAsymm8: + Dequantize<uint8_t>(GetInputTensorData<uint8_t>(0, m_Data), outputData, inputInfo); + break; + case DataType::QuantisedSymm16: + Dequantize<int16_t>(GetInputTensorData<int16_t>(0, m_Data), outputData, inputInfo); + break; + default: + throw InvalidArgumentException("RefDequantizeWorkload: Unsupported input data type"); + } +} + +} // namespace armnn |