COMPMID-2066: Create reference implementation for reflect and symmetric paddings
Change-Id: I18f864b1fa902e5bb0aee8168c02bff73cfd3e59
Signed-off-by: Usama Arif <usama.arif@arm.com>
Reviewed-on: https://review.mlplatform.org/c/874
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
diff --git a/tests/validation/reference/PadLayer.h b/tests/validation/reference/PadLayer.h
index 9406b05..5ebb5ed 100644
--- a/tests/validation/reference/PadLayer.h
+++ b/tests/validation/reference/PadLayer.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2018 ARM Limited.
+ * Copyright (c) 2018-2019 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -39,13 +39,15 @@
/** Reference function to pad an ND tensor. This function is not supposed to be optimized, but to
* clearly and naively execute the padding of a tensor
*
- * @param[in] src Tensor to pad
- * @param[in] paddings Padding size in each dimension
+ * @param[in] src Tensor to pad
+ * @param[in] paddings Padding size in each dimension
+ * @param[in] const_value Constant value to fill padding with
+ * @param[in] mode [optional] Padding mode to use
*
* @return The padded Tensor
*/
template <typename T>
-SimpleTensor<T> pad_layer(const SimpleTensor<T> &src, const PaddingList &paddings);
+SimpleTensor<T> pad_layer(const SimpleTensor<T> &src, const PaddingList &paddings, const PixelValue const_value = PixelValue(), const PaddingMode mode = PaddingMode::CONSTANT);
} // namespace reference
} // namespace validation
} // namespace test