From c18b6ebcfab58c36472be48a9412631ba5440710 Mon Sep 17 00:00:00 2001
From: TianXiaogang <txg4794@163.com>
Date: Fri, 1 Nov 2019 10:14:22 +0800
Subject: [PATCH] fix: fix conv_direct && test=develop (#2314) (#2324)

cherry-pick
---
 lite/kernels/arm/conv_direct.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lite/kernels/arm/conv_direct.h b/lite/kernels/arm/conv_direct.h
index 99025e2cec..24c934e14b 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];
-- 
GitLab