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

[Core] only pass relevant inputs to SDP

parent 596761b8
No related branches found
No related tags found
No related merge requests found
......@@ -51,7 +51,8 @@ def execute_node(node, context, graph):
if node.op_type == "StreamingDataflowPartition":
sdp_node = getCustomOp(node)
model = ModelWrapper(sdp_node.get_nodeattr("model"))
ret = execute_onnx(model, context, True)
inp_ctx = dict(filter(lambda x: x[0] in node.input, context.items()))
ret = execute_onnx(model, inp_ctx, False)
context.update(ret)
else:
if node.domain == "finn":
......
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