diff options
| author | Mohamed Nour Abouelseoud <mohamednour.abouelseoud@arm.com> | 2018-10-18 12:26:19 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-22 16:57:54 +0100 |
| commit | dd6aceaa884815e68ed69fca71de81babd3204da (patch) | |
| tree | 83a0e636a84833e7fd3e78b808d269f80d01a0f8 /src/backends/reference/workloads/Pad.hpp | |
| parent | d134093a271b60e248942af9757e8236e8f41ac1 (diff) | |
| download | armnn-dd6aceaa884815e68ed69fca71de81babd3204da.tar.gz | |
IVGCVSW-2013 Add a UInt8 Reference Implementation for the PAD Operator
Change-Id: I41f3606198db1fda8d72aaf5169594ba9156eb38
Diffstat (limited to 'src/backends/reference/workloads/Pad.hpp')
| -rw-r--r-- | src/backends/reference/workloads/Pad.hpp | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/backends/reference/workloads/Pad.hpp b/src/backends/reference/workloads/Pad.hpp index ed80ef8eb0..42318d6fcf 100644 --- a/src/backends/reference/workloads/Pad.hpp +++ b/src/backends/reference/workloads/Pad.hpp @@ -12,9 +12,10 @@ namespace armnn { +template <typename T> void Pad(const TensorInfo& inputInfo, - const TensorInfo& outputInfo, - std::vector<std::pair<unsigned int, unsigned int>> m_PadList, - const float* inputData, - float* outData); + const TensorInfo& outputInfo, + std::vector<std::pair<unsigned int, unsigned int>> m_PadList, + const T* inputData, + T* outData); } //namespace armnn |