Go
pytorch
29b90b4f
- [functorch] Added conv2d batching rule (pytorch/functorch#10)
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
2 years ago
[functorch] Added conv2d batching rule (pytorch/functorch#10) * Added conv2d batching rule * resolve some comments * added some stuff to conv * remove debugging stuff * remove clamp + add squeeze to vmap failures * Added more conv batching rules
Author
Chillee
Committer
bigfootjon
Parents
354e79af
Loading