aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/Pad.hpp
diff options
context:
space:
mode:
authorDavid Monahan <david.monahan@arm.com>2019-06-19 11:47:21 +0100
committerMatthew Bentham <matthew.bentham@arm.com>2019-06-24 23:09:02 +0000
commit34757810f8b734f5f59485a542b56934ad4cc5f0 (patch)
treec9859e1eb90fdf1e3ba4fbeadb3952cac3477277 /src/backends/reference/workloads/Pad.hpp
parent84da38b0f11ca3db0a439e510514be780f3933ff (diff)
downloadarmnn-34757810f8b734f5f59485a542b56934ad4cc5f0.tar.gz
IVGCVSW-3235 Add scalar to use as padding value in Reference Pad
Signed-off-by: David Monahan <david.monahan@arm.com> Change-Id: If050f318fcb7626bbfae1b8737a1d232a4a5a915
Diffstat (limited to 'src/backends/reference/workloads/Pad.hpp')
-rw-r--r--src/backends/reference/workloads/Pad.hpp5
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/Pad.hpp b/src/backends/reference/workloads/Pad.hpp
index 42318d6fcf..429718596e 100644
--- a/src/backends/reference/workloads/Pad.hpp
+++ b/src/backends/reference/workloads/Pad.hpp
@@ -15,7 +15,8 @@ namespace armnn
template <typename T>
void Pad(const TensorInfo& inputInfo,
const TensorInfo& outputInfo,
- std::vector<std::pair<unsigned int, unsigned int>> m_PadList,
+ std::vector<std::pair<unsigned int, unsigned int>> m_padList,
const T* inputData,
- T* outData);
+ T* outData,
+ const float padValue);
} //namespace armnn