From dd6aceaa884815e68ed69fca71de81babd3204da Mon Sep 17 00:00:00 2001
From: Mohamed Nour Abouelseoud <mohamednour.abouelseoud@arm.com>
Date: Thu, 18 Oct 2018 12:26:19 +0100
Subject: IVGCVSW-2013 Add a UInt8 Reference Implementation for the PAD
 Operator

Change-Id: I41f3606198db1fda8d72aaf5169594ba9156eb38
---
 src/backends/reference/workloads/Pad.hpp | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

(limited to 'src/backends/reference/workloads/Pad.hpp')

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
-- 
cgit v1.2.1