We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7471eb1 commit 2e3c21bCopy full SHA for 2e3c21b
backends/nxp/tests/executors.py
@@ -52,6 +52,13 @@ def inference(
52
return output.detach().numpy()
53
elif isinstance(output, tuple) and len(output) == 1:
54
return output[0].detach().numpy()
55
+ elif isinstance(output, tuple):
56
+ output_names = self.edge_program.graph_signature.user_outputs
57
+
58
+ return {
59
+ name: tensor.detach().numpy()
60
+ for (name, tensor) in zip(output_names, output)
61
+ }
62
63
raise RuntimeError(
64
"Edge program inference with multiple outputs not implemented"
0 commit comments