[luci/export] Split exportModule method (#13864)

This will split exportModule method to prepare two pass execution. ONE-DCO-1.0-Signed-off-by: SaeHie Park <[email protected]>
Samsung · Sep 1, 2024 · d96ddd3 · d96ddd3
1 parent e0d6140
commit d96ddd3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/compiler/luci/export/src/CircleExporterImpl.cpp b/compiler/luci/export/src/CircleExporterImpl.cpp
@@ -159,6 +159,11 @@ void CircleExporterImpl::exportModule(Module *module)
   // prepare model data
   prepareModelData(_builder, md);
 
+  exportModuleData(module, md);
+}
+
+void CircleExporterImpl::exportModuleData(Module *module, SerializedModelData &md)
+{
   std::vector<flatbuffers::Offset<circle::SubGraph>> subgraph_vec;
 
   for (size_t g = 0; g < module->size(); ++g)

diff --git a/compiler/luci/export/src/CircleExporterImpl.h b/compiler/luci/export/src/CircleExporterImpl.h
@@ -64,6 +64,11 @@ class CircleExporterImpl
    */
   void exportModule(Module *module);
 
+  /**
+   * @brief implementation that writes Module into internal buffer
+   */
+  void exportModuleData(Module *module, SerializedModelData &md);
+
 private:
   flatbuffers::FlatBufferBuilder _builder;
 };