From 172035864c8eb73fc46aeec1075423526a768e83 Mon Sep 17 00:00:00 2001 From: Giorgio Arena Date: Fri, 2 Aug 2019 16:00:41 +0100 Subject: COMPMID-2336 Extend validation for depthwise native and fix same pad calculator Change-Id: I9f5cc95bc0cbd94869ac13064ffa0aa0f52a7a61 Signed-off-by: Giorgio Arena Reviewed-on: https://review.mlplatform.org/c/1684 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Reviewed-by: Gian Marco Iodice --- arm_compute/core/Utils.h | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'arm_compute/core/Utils.h') diff --git a/arm_compute/core/Utils.h b/arm_compute/core/Utils.h index b0e26328ed..bc461e7ba9 100644 --- a/arm_compute/core/Utils.h +++ b/arm_compute/core/Utils.h @@ -839,10 +839,12 @@ inline void permute_strides(Dimensions &dimensions, const PermutationVector & * @param[in] conv_info Convolution information (containing strides) * @param[in] data_layout (Optional) Data layout of the input and weights tensor * @param[in] dilation (Optional) Dilation factor used in the convolution. + * @param[in] rounding_type (Optional) Dimension rounding type when down-scaling. * * @return PadStrideInfo for SAME padding */ -PadStrideInfo calculate_same_pad(TensorShape input_shape, TensorShape weights_shape, PadStrideInfo conv_info, DataLayout data_layout = DataLayout::NCHW, const Size2D &dilation = Size2D(1u, 1u)); +PadStrideInfo calculate_same_pad(TensorShape input_shape, TensorShape weights_shape, PadStrideInfo conv_info, DataLayout data_layout = DataLayout::NCHW, const Size2D &dilation = Size2D(1u, 1u), + const DimensionRoundingType &rounding_type = DimensionRoundingType::FLOOR); /** Returns expected width and height of the deconvolution's output tensor. * -- cgit v1.2.1