pytorch
a7e7fbab - Add shape functions for conv_transpose2d.input and convolution (#80860)

Commit
3 years ago
Add shape functions for conv_transpose2d.input and convolution (#80860) As @silvasean requested in [this issue](https://github.com/llvm/torch-mlir/pull/917#discussion_r896154545) here is the shape code from Torch-MLIR for conv_transpose2d.input and convolution (updated for the transposed case). Pull Request resolved: https://github.com/pytorch/pytorch/pull/80860 Approved by: https://github.com/Gamrix
Author
Committer
Parents
Loading