Skip to content

Commit 60ede6e

Browse files
committed
Add support for EdgeProgram test inference with multiple outputs
1 parent 55ccc62 commit 60ede6e

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

backends/nxp/tests/executors.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,10 @@ def inference(self, input_data: Union[numpy.ndarray, Dict[int, numpy.ndarray]])
4545
return output.detach().numpy()
4646
elif isinstance(output, tuple) and len(output) == 1:
4747
return output[0].detach().numpy()
48+
elif isinstance(output, tuple):
49+
output_names = self.edge_program.graph_signature.user_outputs
50+
51+
return {name: tensor.detach().numpy() for (name, tensor) in zip(output_names, output)}
4852

4953
raise RuntimeError("Edge program inference with multiple outputs not implemented")
5054

0 commit comments

Comments
 (0)