pytorch
7a3b05ea - Fix hardswish inplace op for strided tensor with skipped elements (#61622)

Commit
3 years ago
Fix hardswish inplace op for strided tensor with skipped elements (#61622) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61622 Hardswish inplace op would return incorrect response for strided tensor inputs that skip elements like a slice. Create a contiguous tensor and copy elements back to return the correct answer Test Plan: Internal CI tests Reviewed By: kimishpatel Differential Revision: D29689745 fbshipit-source-id: 11618a8d865f550f6b70637345f9ebc3e5676f11
Author
Akshit Khurana
Parents
Loading