diff --git a/lite/kernels/arm/conv_direct.h b/lite/kernels/arm/conv_direct.h
index 99025e2cec13660d96c2c71481ed3a52a4332ea9..24c934e14b5ae5e7de4d089da1611deb0e77fefb 100644
--- a/lite/kernels/arm/conv_direct.h
+++ b/lite/kernels/arm/conv_direct.h
@@ -156,6 +156,7 @@ class DirectConv : public KernelLite<TARGET(kARM), Ptype> {
     auto x_dims = param.x->dims();
     auto w_dims = param.filter->dims();
     auto o_dims = param.output->dims();
+    last_shape_ = x_dims;
 
     int ic = x_dims[1];
     int oc = o_dims[1];