diff --git a/lite/kernels/xpu/slice_compute.cc b/lite/kernels/xpu/slice_compute.cc index 9e131aa4b49..6353d6114e6 100644 --- a/lite/kernels/xpu/slice_compute.cc +++ b/lite/kernels/xpu/slice_compute.cc @@ -172,6 +172,7 @@ void SliceCompute::Run() { int end = ends[i] < 0 ? ends[i] + static_cast(in_dims[axis]) : ends[i]; x_dim_end_[axis] = (std::min)(end, static_cast(in_dims[axis])); } + out->Resize(out_dims); int r = xdnn::slice(ctx.GetRawContext(), /* context */