From e0733f3ebb23fd6cee7d8102818427629baa2405 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Thu, 6 Aug 2020 01:14:27 +0200
Subject: [PATCH] [Core] support renamed tensors for SDP output
---
src/finn/core/onnx_exec.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/src/finn/core/onnx_exec.py b/src/finn/core/onnx_exec.py
index 6c2bf6d05..0c01a48a0 100644
--- a/src/finn/core/onnx_exec.py
+++ b/src/finn/core/onnx_exec.py
@@ -60,7 +60,11 @@ def execute_node(node, context, graph):
inp_ctx[new_iname] = inp_ctx[old_iname]
del inp_ctx[old_iname]
ret = execute_onnx(model, inp_ctx, False)
- context.update(ret)
+ # output may have been renamed in partition
+ assert len(ret) == 1
+ node_oname = node.output[0]
+ model_oname = model.graph.output[0].name
+ context[node_oname] = ret[model_oname]
else:
if node.domain == "finn":
--
GitLab