pytorch
01a88625 - OpInfo tests for `nn.functional.max_pool{n}d`. (#68075)

Commit
4 years ago
OpInfo tests for `nn.functional.max_pool{n}d`. (#68075) Summary: As per title. It is planned to use these tests for fixing issues with the max_unpools' backward methods reported in https://github.com/pytorch/pytorch/issues/67658 and https://github.com/pytorch/pytorch/issues/67657. max_unpool.backward methods are not tested and implemented with custom kernels. We can replace these kernels with advanced indexing operations (i.e. `gather`) which are efficient and well tested. Pull Request resolved: https://github.com/pytorch/pytorch/pull/68075 Reviewed By: malfet Differential Revision: D32308317 Pulled By: mruberry fbshipit-source-id: 9f91c6e6a9d78c19230e93fc0a3164f4eb7b8ec5
Author
Parents
Loading