From b4af2c6738614850aaca3754904f0e8e3b17f0b2 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Mon, 10 Dec 2018 18:45:35 +0000 Subject: COMPMID-1710: Fixes in StrideSlice calculations. Change-Id: I66eb922f1ff15142de278bf4439a61c979f98ba7 Reviewed-on: https://review.mlplatform.org/382 Reviewed-by: Matthew Bentham Tested-by: Arm Jenkins Reviewed-by: Pablo Marquez --- tests/validation/reference/SliceOperations.cpp | 28 ++++++++++++++------------ 1 file changed, 15 insertions(+), 13 deletions(-) (limited to 'tests/validation/reference/SliceOperations.cpp') diff --git a/tests/validation/reference/SliceOperations.cpp b/tests/validation/reference/SliceOperations.cpp index 04b5b98453..40ca9de927 100644 --- a/tests/validation/reference/SliceOperations.cpp +++ b/tests/validation/reference/SliceOperations.cpp @@ -24,6 +24,7 @@ #include "SliceOperations.h" #include "arm_compute/core/utils/helpers/tensor_transform.h" +#include "arm_compute/core/utils/misc/ShapeCalculator.h" namespace arm_compute { @@ -50,11 +51,8 @@ SimpleTensor slice(const SimpleTensor &src, Coordinates starts, Coordinate // Get source shape const TensorShape &src_shape = src.shape(); - // Get actual end - Coordinates ends_abs = slice_absolute_end_coords(src_shape, ends); - // Get destination shape - TensorShape dst_shape = compute_slice_output_shape(src_shape, starts, ends_abs); + TensorShape dst_shape = arm_compute::misc::shape_calculator::compute_slice_shape(src_shape, starts, ends); // Create destination tensor SimpleTensor dst{ dst_shape, src.data_type(), 1 }; @@ -98,20 +96,24 @@ SimpleTensor strided_slice(const SimpleTensor &src, // Get source shape const TensorShape &src_shape = src.shape(); - // Get actual start, end coordinates and strides - const Coordinates final_strides = strided_slice_strides(src_shape, strides); - const Coordinates starts_abs = strided_slice_absolute_start_coords(src_shape, starts, final_strides, begin_mask); - const Coordinates ends_abs = strided_slice_absolute_end_coords(src_shape, starts_abs, ends, final_strides, end_mask, shrink_axis_mask); - // Get destination shape - const TensorShape dst_shape = compute_strided_slice_output_shape(src_shape, starts_abs, ends_abs, final_strides); + const TensorShape dst_shape = compute_strided_slice_output_shape(src_shape, starts, ends, strides, begin_mask, end_mask, shrink_axis_mask); // Create destination tensor SimpleTensor dst{ dst_shape, src.data_type(), 1 }; + // Get coordinates + Coordinates starts_abs, ends_abs, final_strides; + std::tie(starts_abs, ends_abs, final_strides) = calculate_strided_slice_coords(src_shape, + starts, ends, strides, + begin_mask, end_mask, shrink_axis_mask); + // Perform strided slice - Window win; - win.use_tensor_dimensions(dst_shape); + unsigned int idx = 0; + Window win; + win.use_tensor_dimensions(compute_strided_slice_output_shape(src_shape, + starts, ends, strides, + begin_mask, end_mask, shrink_axis_mask, true)); execute_window_loop(win, [&](const Coordinates & id) { Coordinates offset; @@ -119,7 +121,7 @@ SimpleTensor strided_slice(const SimpleTensor &src, { offset.set(i, starts_abs[i] + id[i] * final_strides[i]); } - *reinterpret_cast(dst(id)) = *reinterpret_cast(src(offset)); + dst.data()[idx++] = *reinterpret_cast(src(offset)); }); return dst; -- cgit v1.2.1