pytorch
956bafef
- [onnx export] Add broadcast to matmul shape inference (#70534)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
[onnx export] Add broadcast to matmul shape inference (#70534) Reuse the same broadcast code from the function `ProcessBroadcastNode`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/72990
References
#73443 - merge master into lazy_tensor_staging
Author
BowenBao
Committer
pytorchmergebot
Parents
98f9ff90
Loading