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 0a27e93 commit ccc6a0bCopy full SHA for ccc6a0b
backends/nxp/tests/executors.py
@@ -46,6 +46,13 @@ def inference(
46
return output.detach().numpy()
47
elif isinstance(output, tuple) and len(output) == 1:
48
return output[0].detach().numpy()
49
+ elif isinstance(output, tuple):
50
+ output_names = self.edge_program.graph_signature.user_outputs
51
+
52
+ return {
53
+ name: tensor.detach().numpy()
54
+ for (name, tensor) in zip(output_names, output)
55
+ }
56
57
raise RuntimeError(
58
"Edge program inference with multiple outputs not implemented"
0 commit comments