[luci/export] Add keep_num_dims to CircleFullyConnected (Samsung#8451)

This commit adds `keep_num_dims` to `CircleFullyConnected`. ONE-DCO-1.0-Signed-off-by: Seok NamKoong <[email protected]>
rnshah9 · Feb 18, 2022 · 65bbd1c · 65bbd1c
1 parent 8f55147
commit 65bbd1c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/compiler/luci/export/src/CircleBuiltinTypesExtractor.h b/compiler/luci/export/src/CircleBuiltinTypesExtractor.h
@@ -170,9 +170,9 @@ class BuiltinOptionsExtractor final
   }
   flatbuffers::Offset<void> visit(luci::CircleFullyConnected *node)
   {
-    return circle::CreateFullyConnectedOptions(_builder,
-                                               to_circle_actfunc(node->fusedActivationFunction()),
-                                               to_circle_weightsformat(node->weights_format()))
+    return circle::CreateFullyConnectedOptions(
+             _builder, to_circle_actfunc(node->fusedActivationFunction()),
+             to_circle_weightsformat(node->weights_format()), node->keep_num_dims())
       .Union();
   }
   flatbuffers::Offset<void> visit(luci::CircleGather *node)