提交 e5e3fdf3 编写于 作者: H Haitao

fix the issue for common fc_blas to handle input less than 4 dims


Former-commit-id: 3d5524cd33e6fa230a030cca9f433c33aaec236f
上级 205e7e24
......@@ -67,8 +67,15 @@ struct FcBlasOps : public NodeOps
int batch_number = in_dims[0];
int inc = in_dims[1];
int inh = in_dims[2];
int inw = in_dims[3];
int inh =1;
int inw =1;
if(in_dims.size()>2)
inh=in_dims[2];
if(in_dims.size()>3)
inw=in_dims[3];
int in_chw = inc * inh * inw;
/* specially handling on tensorflow models */
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册