提交 2275ea48 编写于 作者: D dingminghui 提交者: jackzhang235

fix(fc_op): fix fc_op to support n > 1

上级 78f44bfc
......@@ -50,7 +50,7 @@ int FCConverter(void* ctx, OpLite* op, KernelBase* kernel) {
auto input_scale = op_info->GetAttr<float>("input_scale");
std::vector<int64_t> output_shape_nhwc({1, 1, 1, w_dims[1]});
std::vector<int64_t> output_shape_nhwc({x_dims[0], 1, 1, w_dims[1]});
auto output_tensor = graph->AddNode(output_var_name,
output_shape_nhwc,
CNML_TENSOR,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册