diff options
Diffstat (limited to 'src/cpu/kernels/CpuDepthwiseConv2dNativeKernel.cpp')
-rw-r--r-- | src/cpu/kernels/CpuDepthwiseConv2dNativeKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/cpu/kernels/CpuDepthwiseConv2dNativeKernel.cpp b/src/cpu/kernels/CpuDepthwiseConv2dNativeKernel.cpp index d79fe87d1b..d09cc1debf 100644 --- a/src/cpu/kernels/CpuDepthwiseConv2dNativeKernel.cpp +++ b/src/cpu/kernels/CpuDepthwiseConv2dNativeKernel.cpp @@ -78,7 +78,7 @@ struct DepthwiseConvolutionRunInfo x_start(w.x().start()), x_end(w.x().end()), x_step(static_cast<uint32_t>(num_read_elements_per_iteration * depth_multiplier)), - x_leftover_start(std::max(static_cast<int32_t>(w.x().end()) - static_cast<int32_t>(x_step) + 1, int32_t(0))), + x_leftover_start(std::max(static_cast<int32_t>(w.x().end() + 1) - static_cast<int32_t>(x_step), int32_t(0))), input_stride_y(input.strides_in_bytes().y()), input_stride_z(input.strides_in_bytes().z()), input_max_offset(input.strides_in_bytes().z() * input.dimension(height_idx) - (input.padding().bottom + input.padding().top) * input.strides_in_bytes().y()), |