diff options
author | Giorgio Arena <giorgio.arena@arm.com> | 2019-08-02 16:00:41 +0100 |
---|---|---|
committer | Giorgio Arena <giorgio.arena@arm.com> | 2019-08-08 10:20:13 +0000 |
commit | 172035864c8eb73fc46aeec1075423526a768e83 (patch) | |
tree | 620fc0f0c8ae69146fe577dc67ae3dbe95d8bc46 /arm_compute/core | |
parent | fed275d76d8322d51872845378adc0058c02bfc1 (diff) | |
download | ComputeLibrary-172035864c8eb73fc46aeec1075423526a768e83.tar.gz |
COMPMID-2336 Extend validation for depthwise native and fix same pad calculator
Change-Id: I9f5cc95bc0cbd94869ac13064ffa0aa0f52a7a61
Signed-off-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1684
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Diffstat (limited to 'arm_compute/core')
-rw-r--r-- | arm_compute/core/Utils.h | 4 |
1 files changed, 3 insertions, 1 deletions
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<T> &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. * |