diff --git a/compiler/luci/export/src/CircleExporterImpl.cpp b/compiler/luci/export/src/CircleExporterImpl.cpp index 00e7ace2c99..de26b379cce 100644 --- 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> 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 index 5911fadba43..0bcd44885bd 100644 --- 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; };