Skip to content
Snippets Groups Projects
Commit ed0ceba3 authored by Maxence Naud's avatar Maxence Naud
Browse files

Merge branch 'dev' into 'master'

version 0.1.0

See merge request eclipse/aidge/aidge_export_cpp!6
parents b5213340 42e7e0d1
No related branches found
No related tags found
No related merge requests found
......@@ -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()
)
......
......@@ -446,7 +446,7 @@ class FcCPP(ExportNode):
list_actions.append(generate_action(
KERNELS_FORWARD.FC,
name=self.name,
input_name=self.inputs[0].name(),
input_name=self.inputs[0].name() if self.inputs[0] else self.name + "_input",
output_name=self.name,
weights_name=self.inputs[1].name(),
biases_name=self.inputs[2].name()
......
0.0.1
\ No newline at end of file
0.1.0
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment