Skip to content
Snippets Groups Projects

version 0.1.0

Merged Maxence Naud requested to merge dev into master
3 files
+ 4
4
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -44,7 +44,7 @@ def export(export_folder, graphview, scheduler):
list_actions = []
list_configs = []
list_forward_nodes = scheduler.get_static_scheduling()
list_forward_nodes = [i for i in scheduler.get_static_scheduling() if i.type() != "Producer"]
list_op = {}
for node in graphview.get_nodes():
@@ -87,7 +87,7 @@ def export(export_folder, graphview, scheduler):
headers=list_configs,
actions=list_actions,
input_t="float",
inputs=list_forward_nodes[0].get_parents()[0].name(),
inputs= list_forward_nodes[0].name()+"_input" if list_forward_nodes[0].get_parents()[0] is None else list_forward_nodes[0].get_parents()[0].name(),
output_t="float",
outputs=list_forward_nodes[-1].name()
)
Loading