[Cherry-Pick][HuaweiKirinNPU] Fix the converter of fc op for Harmony …

…OS (PaddlePaddle#8326)
wangleilei001 · Jan 22, 2022 · 07e5d99 · 07e5d99
1 parent a1617bb
commit 07e5d99
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lite/kernels/npu/bridges/fc_op.cc b/lite/kernels/npu/bridges/fc_op.cc
@@ -82,6 +82,7 @@ int FCConverter(void* ctx, OpLite* op, KernelBase* kernel) {
   auto fc_op = fc_node->data<ge::op::FullConnection>();
   fc_op->set_input_x(*reshaped_input_node->data());
   fc_op->set_input_w(*trans_w_node->data());
+  fc_op->set_attr_num_output(n);
 
   // Add bias node if bias tensor exists
   if (HasInputArg(op_info, scope, "Bias")) {