pytorch
4a762cb6 - [MPS] Fix channels last copies in ELU,ReLU and Hardswish (#94664)

Commit
1 year ago
[MPS] Fix channels last copies in ELU,ReLU and Hardswish (#94664) Fixes test_modules.py tests: ``` test_memory_format_nn_Hardswish_mps_float32 test_non_contiguous_tensors_nn_Hardswish_mps_float32 test_memory_format_nn_ReLU_mps_float32 ``` Fixes elu when ran with `ChannelsLast` memory format. Pull Request resolved: https://github.com/pytorch/pytorch/pull/94664 Approved by: https://github.com/kulinseth
Author
Committer
Parents
Loading