diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-07-08 08:57:17 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-07-09 10:48:58 +0000 |
commit | e6eaf661c5b84f4ca051daaf08281d9b8de3fcb9 (patch) | |
tree | 63f3fa42b5a0389b9e69940d0960ec7fa609af93 /src/backends/reference/workloads | |
parent | 6059784511ce472c9df2289582ad13a0e2aa160e (diff) | |
download | armnn-e6eaf661c5b84f4ca051daaf08281d9b8de3fcb9.tar.gz |
IVGCVSW-3399 Add support of QuantisedSymm16 to Pad
* Add support of QuantisedSymm16 to Pad
* Fix custom padding tests
* Add unit tests for Pad QuantisedSymm16
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ieb9a4492760b13fa030a42eb4d38fd9ffd7a8c76
Diffstat (limited to 'src/backends/reference/workloads')
-rw-r--r-- | src/backends/reference/workloads/Pad.cpp | 6 | ||||
-rw-r--r-- | src/backends/reference/workloads/RefPadWorkload.cpp | 1 | ||||
-rw-r--r-- | src/backends/reference/workloads/RefPadWorkload.hpp | 3 |
3 files changed, 9 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/Pad.cpp b/src/backends/reference/workloads/Pad.cpp index 1e58124627..41435f47d2 100644 --- a/src/backends/reference/workloads/Pad.cpp +++ b/src/backends/reference/workloads/Pad.cpp @@ -175,5 +175,11 @@ template void Pad<uint8_t>(const TensorInfo& inputInfo, const uint8_t* inputData, uint8_t* outData, const float padValue); +template void Pad<int16_t>(const TensorInfo& inputInfo, + const TensorInfo& outputInfo, + std::vector<std::pair<unsigned int, unsigned int>> m_PadList, + const int16_t* inputData, + int16_t* outData, + const float padValue); } //namespace armnn
\ No newline at end of file diff --git a/src/backends/reference/workloads/RefPadWorkload.cpp b/src/backends/reference/workloads/RefPadWorkload.cpp index e9724c449f..5e59d83dc9 100644 --- a/src/backends/reference/workloads/RefPadWorkload.cpp +++ b/src/backends/reference/workloads/RefPadWorkload.cpp @@ -35,5 +35,6 @@ void RefPadWorkload<DataType>::Execute() const template class RefPadWorkload<DataType::Float32>; template class RefPadWorkload<DataType::QuantisedAsymm8>; +template class RefPadWorkload<DataType::QuantisedSymm16>; } //namespace armnn
\ No newline at end of file diff --git a/src/backends/reference/workloads/RefPadWorkload.hpp b/src/backends/reference/workloads/RefPadWorkload.hpp index 8c6d01351b..b1de53e930 100644 --- a/src/backends/reference/workloads/RefPadWorkload.hpp +++ b/src/backends/reference/workloads/RefPadWorkload.hpp @@ -31,6 +31,7 @@ public: }; using RefPadFloat32Workload = RefPadWorkload<DataType::Float32>; -using RefPadUint8Workload = RefPadWorkload<DataType::QuantisedAsymm8>; +using RefPadQAsymm8Workload = RefPadWorkload<DataType::QuantisedAsymm8>; +using RefPadQSymm16Workload = RefPadWorkload<DataType::QuantisedSymm16>; } //namespace armnn |