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 c90f4d9 commit 34c1569Copy full SHA for 34c1569
backends/nxp/tests/executors.py
@@ -57,6 +57,13 @@ def inference(
57
return output.detach().numpy()
58
elif isinstance(output, tuple) and len(output) == 1:
59
return output[0].detach().numpy()
60
+ elif isinstance(output, tuple):
61
+ output_names = self.edge_program.graph_signature.user_outputs
62
+
63
+ return {
64
+ name: tensor.detach().numpy()
65
+ for (name, tensor) in zip(output_names, output)
66
+ }
67
68
raise RuntimeError(
69
"Edge program inference with multiple outputs not implemented"
0 commit comments