diff options
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 |