Skip to content

Fix ONNX input naming#97

Open
markuswt wants to merge 1 commit into
jiazhihao:masterfrom
markuswt:fix-onnx-inputs
Open

Fix ONNX input naming#97
markuswt wants to merge 1 commit into
jiazhihao:masterfrom
markuswt:fix-onnx-inputs

Conversation

@markuswt

@markuswt markuswt commented Jan 26, 2023

Copy link
Copy Markdown

Currently, all input tensors in generated ONNX models are labeled "data". Since ONNX supports multiple inputs, this creates conflicts. For example, the following graph would be exported as having two separate inputs:

g = taso.new_graph()
i = g.new_input(dims=(5, 5, 10, 10))
g.add(i, i)

This PR fixes this issue by giving all inputs a unique name.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant