diff options
author | Isabella Gottardi <isabella.gottardi@arm.com> | 2019-01-30 15:45:42 +0000 |
---|---|---|
committer | Isabella Gottardi <isabella.gottardi@arm.com> | 2019-02-01 17:02:45 +0000 |
commit | 6a91440273444b14d158056b5c199e59ca0051e5 (patch) | |
tree | 95f82ddf4909d192f1cb70560a30bd4c11346281 /arm_compute | |
parent | b7c308a1ad32af4198dcc7eaa73f44fef27dc8fc (diff) | |
download | ComputeLibrary-6a91440273444b14d158056b5c199e59ca0051e5.tar.gz |
COMPMID-1710: Allow NHWC datalayout in SAME pad calculation
Change-Id: Id3788772ad62c7e2d962bc2cb8812b9503e2e836
Signed-off-by: Isabella Gottardi <isabella.gottardi@arm.com>
Reviewed-on: https://review.mlplatform.org/603
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: VidhyaSudhan Loganathan <vidhyasudhan.loganathan@arm.com>
Diffstat (limited to 'arm_compute')
-rw-r--r-- | arm_compute/core/Utils.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/arm_compute/core/Utils.h b/arm_compute/core/Utils.h index 816999db51..d112259d39 100644 --- a/arm_compute/core/Utils.h +++ b/arm_compute/core/Utils.h @@ -827,10 +827,11 @@ inline void permute_strides(Dimensions<T> &dimensions, const PermutationVector & * @param[in] input_shape Input shape * @param[in] weights_shape Weights shape * @param[in] conv_info Convolution information (containing strides) + * @param[in] data_layout (Optional) Data layout of the input and weights tensor * * @return PadStrideInfo for SAME padding */ -PadStrideInfo calculate_same_pad(TensorShape input_shape, TensorShape weights_shape, PadStrideInfo conv_info); +PadStrideInfo calculate_same_pad(TensorShape input_shape, TensorShape weights_shape, PadStrideInfo conv_info, DataLayout data_layout = DataLayout::NCHW); /** Returns expected width and height of the deconvolution's output tensor. * |