diff options
-rw-r--r-- | src/core/NEON/kernels/NEScaleKernel.cpp | 2 | ||||
-rw-r--r-- | tests/validation/NEON/Scale.cpp | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/src/core/NEON/kernels/NEScaleKernel.cpp b/src/core/NEON/kernels/NEScaleKernel.cpp index 8e9a34637b..a2a44fca18 100644 --- a/src/core/NEON/kernels/NEScaleKernel.cpp +++ b/src/core/NEON/kernels/NEScaleKernel.cpp @@ -377,7 +377,7 @@ void NEScaleKernel::configure(const ITensor *input, const ITensor *dx, const ITe // Add constant border only on top in case of NHWC layout if(data_layout == DataLayout::NHWC) { - _border_size = (border_mode == BorderMode::CONSTANT && policy == InterpolationPolicy::BILINEAR && use_padding) ? BorderSize(1, 0, 0, 0) : BorderSize(0); + _border_size = (border_mode != BorderMode::REPLICATE && policy == InterpolationPolicy::BILINEAR && use_padding) ? BorderSize(1, 0, 0, 0) : BorderSize(0); } // Area interpolation behaves as Nearest Neighbour in case of up-sampling diff --git a/tests/validation/NEON/Scale.cpp b/tests/validation/NEON/Scale.cpp index c05b8ac03b..d419c62651 100644 --- a/tests/validation/NEON/Scale.cpp +++ b/tests/validation/NEON/Scale.cpp @@ -183,7 +183,7 @@ DATA_TEST_CASE(Configuration, framework::DatasetMode::ALL, combine(combine(combi if(data_layout == DataLayout::NHWC) { read_padding = calculator.required_padding(PaddingCalculator::Option::EXCLUDE_BORDER); - if(border_mode == BorderMode::CONSTANT && policy == InterpolationPolicy::BILINEAR) + if(border_mode != BorderMode::REPLICATE && policy == InterpolationPolicy::BILINEAR) { read_padding.top = 1; } |