aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/Pad.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/Pad.hpp')
-rw-r--r--src/backends/reference/workloads/Pad.hpp12
1 files changed, 5 insertions, 7 deletions
diff --git a/src/backends/reference/workloads/Pad.hpp b/src/backends/reference/workloads/Pad.hpp
index 429718596e..e7be44e88c 100644
--- a/src/backends/reference/workloads/Pad.hpp
+++ b/src/backends/reference/workloads/Pad.hpp
@@ -5,18 +5,16 @@
#pragma once
-#include "armnn/DescriptorsFwd.hpp"
#include "armnn/Tensor.hpp"
-#include <vector>
+#include <backendsCommon/Workload.hpp>
+#include <backendsCommon/WorkloadData.hpp>
namespace armnn
{
-template <typename T>
+
void Pad(const TensorInfo& inputInfo,
const TensorInfo& outputInfo,
- std::vector<std::pair<unsigned int, unsigned int>> m_padList,
- const T* inputData,
- T* outData,
- const float padValue);
+ const PadQueueDescriptor& data);
+
} //namespace armnn