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