diff options
Diffstat (limited to 'tests/validation/Helpers.h')
| -rw-r--r-- | tests/validation/Helpers.h | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tests/validation/Helpers.h b/tests/validation/Helpers.h index 00e588e7b7..a8804ad7e7 100644 --- a/tests/validation/Helpers.h +++ b/tests/validation/Helpers.h @@ -240,6 +240,16 @@ std::pair<int, int> get_symm_quantized_per_channel_bounds(const QuantizationInfo * @note This function adds padding to the input tensors only if data_layout == DataLayout::NHWC */ void add_padding_x(std::initializer_list<ITensor *> tensors, const DataLayout &data_layout = DataLayout::NHWC, bool only_right_pad = false); + +/** Add random padding along the Y axis (between 1 and 4 rows per side) to all the input tensors. + * This is used in our validation suite in order to simulate implicit padding addition after configuring, but before allocating. + * + * @param[in] tensors List of tensors to add padding to + * @param[in] data_layout (Optional) Data layout of the operator + * + * @note This function adds padding to the input tensors only if data_layout == DataLayout::NHWC + */ +void add_padding_y(std::initializer_list<ITensor *> tensors, const DataLayout &data_layout = DataLayout::NHWC); } // namespace validation } // namespace test } // namespace arm_compute |