diff options
Diffstat (limited to 'src/core')
| -rw-r--r-- | src/core/NEON/kernels/NEScaleKernel.cpp | 2 | ||||
| -rw-r--r-- | src/core/utils/helpers/tensor_transform.cpp | 9 |
2 files changed, 9 insertions, 2 deletions
diff --git a/src/core/NEON/kernels/NEScaleKernel.cpp b/src/core/NEON/kernels/NEScaleKernel.cpp index a2a44fca18..80da54f50a 100644 --- a/src/core/NEON/kernels/NEScaleKernel.cpp +++ b/src/core/NEON/kernels/NEScaleKernel.cpp @@ -227,7 +227,7 @@ inline void scale_bilinear_nhwc_core(const ITensor *input, const ITensor *offset border_value = static_cast<T>(constant_border_value.get<ConstType>()); } - auto is_valid = [](int x, int low_x, int high_x, int y, int low_y, int high_y) + auto is_valid = [](int64_t x, int64_t low_x, int64_t high_x, int64_t y, int64_t low_y, int64_t high_y) { return !(x < low_x || x > high_x || y < low_y || y > high_y); }; diff --git a/src/core/utils/helpers/tensor_transform.cpp b/src/core/utils/helpers/tensor_transform.cpp index f6a54a59ee..cd874b24b3 100644 --- a/src/core/utils/helpers/tensor_transform.cpp +++ b/src/core/utils/helpers/tensor_transform.cpp @@ -88,7 +88,14 @@ int calculate_end_on_index(TensorShape input_shape, int index, int start_on_inde // Shrink dimension if(shrink_axis) { - stop = start_on_index + 1; + if(start_on_index == std::numeric_limits<int>::max()) + { + stop = start_on_index; + } + else + { + stop = start_on_index + 1; + } } // Reset in case of begin mask present |