Skip to content
Snippets Groups Projects
Commit 4584f0ba authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Test] bugfix in test_set_folding

parent f2b165f4
No related branches found
No related tags found
No related merge requests found
......@@ -98,7 +98,8 @@ def make_multi_fclayer_model(ch, wdt, adt, tdt, nnodes):
model.set_tensor_datatype("outp", adt)
for i in range(1, nnodes + 1):
model.graph.value_info.append(tensors[i])
if tensors[i].name != "outp":
model.graph.value_info.append(tensors[i])
model.set_initializer("weights_" + str(i - 1), W)
model.set_initializer("thresh_" + str(i - 1), T)
model.set_tensor_datatype("weights_" + str(i - 1), wdt)
......
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