17#ifndef LUCI_INTERPRETER_CORE_RUNTIMEMODULE_H
18#define LUCI_INTERPRETER_CORE_RUNTIMEMODULE_H
20#include "core/RuntimeGraph.h"
26#include "core/RuntimeGraph.h"
35#ifdef USE_STATIC_ALLOC
58 friend class training::TrainingModule;
63 _graphs.emplace_back(memory_manager, &_circle_reader,
this,
64 _circle_reader.get_current_subgraph_index());
78 std::vector<BaseRuntimeGraph> _graphs;
83 std::unique_ptr<training::TrainableWeightStorage> _storage;
Loads Circle file and provides helpers to access attributes.
bool select_subgraph(uint32_t subgraph)
BaseRuntimeGraph * getRuntimeGraphAt(uint32_t pos)
BaseRuntimeGraph * getMainGraph() const
void selectSubgraph(uint32_t index)
void addGraph(MemoryManager *memory_manager)
CircleReader & getCircleReader()
RuntimeGraph BaseRuntimeGraph
SimpleMemoryManager MemoryManager