torch.onnx.dynamo_export: Unsupported FX nodes: {'call_function': ['aten.pixel_shuffle.default']} #120078
Labels
module: onnx
Related to torch.onnx
OSS contribution wanted
PR from open source contributors welcome to solve this issue.
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
🐛 Describe the bug
torch.onnx.dynamo_export doesn't work on nn.PixelShuffle
The code above gives the following exception:
Serif report: report_dynamo_export.sarif
Versions
The text was updated successfully, but these errors were encountered: