onnxruntime
01acc25d - [DML EP] Fix the output shapes of nodes with multiple outputs in the graph builder (#20289)

Commit
1 year ago
[DML EP] Fix the output shapes of nodes with multiple outputs in the graph builder (#20289) The graph builder currently doesn't assign the correct shapes for subgraphs that have more than 1 output, and where each output comes from a different node. `nodeOutputShapes` should be a map of shapes (1:1 relationship), and not a map of lists of shapes (1:N relationship) since an output referenced by `arg->Name()` can only have 1 output. Take for example the following example of a subgraph where a node has 2 outputs, then each output feeds into an elementwise op. Both nodes will have a `targetIndex` of 0, and we were using this target index to query their shape, resulting in both outputs querying the same shape. In reality, what we need to do is use the `GraphOutputIndex` ofthe subgraph to query the correct output shape of the subgraph.
Parents
Loading