diff options
-rw-r--r-- | tests/validation/fixtures/ScaleFixture.h | 4 | ||||
-rw-r--r-- | tests/validation/reference/Scale.cpp | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/tests/validation/fixtures/ScaleFixture.h b/tests/validation/fixtures/ScaleFixture.h index 199186f8e2..12d9012ece 100644 --- a/tests/validation/fixtures/ScaleFixture.h +++ b/tests/validation/fixtures/ScaleFixture.h @@ -133,8 +133,8 @@ protected: const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); TensorShape shape_scaled(shape); - shape_scaled.set(idx_width, shape[idx_width] * _scale_x); - shape_scaled.set(idx_height, shape[idx_height] * _scale_y); + shape_scaled.set(idx_width, shape[idx_width] * _scale_x, /* apply_dim_correction = */ false); + shape_scaled.set(idx_height, shape[idx_height] * _scale_y, /* apply_dim_correction = */ false); TensorType dst = create_tensor<TensorType>(shape_scaled, _data_type, 1, _quantization_info, data_layout); // Create and configure function diff --git a/tests/validation/reference/Scale.cpp b/tests/validation/reference/Scale.cpp index 06bc42004f..cb4ad6f909 100644 --- a/tests/validation/reference/Scale.cpp +++ b/tests/validation/reference/Scale.cpp @@ -42,8 +42,8 @@ SimpleTensor<T> scale_core(const SimpleTensor<T> &in, float scale_x, float scale // Add 1 if ceil_policy_scale is true const size_t round_value = ceil_policy_scale ? 1U : 0U; TensorShape shape_scaled(in.shape()); - shape_scaled.set(0, (in.shape()[0] + round_value) * scale_x); - shape_scaled.set(1, (in.shape()[1] + round_value) * scale_y); + shape_scaled.set(0, (in.shape()[0] + round_value) * scale_x, /* apply_dim_correction = */ false); + shape_scaled.set(1, (in.shape()[1] + round_value) * scale_y, /* apply_dim_correction = */ false); SimpleTensor<T> out(shape_scaled, in.data_type()); // Compute the ratio between source width/height and destination width/height |