提交 d045f646 编写于 作者: T TianXiaogang 提交者: cyj1986

fix: fix conv_direct && test=develop (#2314)

上级 1c96b4b8
......@@ -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];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册