diff --git a/lite/kernels/mlu/bridges/fc_op.cc b/lite/kernels/mlu/bridges/fc_op.cc index 43a75daa2b3d2d6200f3607e213ab62ee6ba3cdb..f480a9110790406ddb2aa7464221c7062b26268e 100644 --- a/lite/kernels/mlu/bridges/fc_op.cc +++ b/lite/kernels/mlu/bridges/fc_op.cc @@ -50,7 +50,7 @@ int FCConverter(void* ctx, OpLite* op, KernelBase* kernel) { auto input_scale = op_info->GetAttr("input_scale"); - std::vector output_shape_nhwc({1, 1, 1, w_dims[1]}); + std::vector output_shape_nhwc({x_dims[0], 1, 1, w_dims[1]}); auto output_tensor = graph->AddNode(output_var_name, output_shape_nhwc, CNML_TENSOR,