pytorch
d52d2bd5 - [ROCm] MIOpen fused convolution relu (#82002)

Commit
2 years ago
[ROCm] MIOpen fused convolution relu (#82002) Adds MIOpen fused convolution relu for fp32 and contiguous memory format. Adds fallbacks for conv + z + bias + relu, fp16, and channels last until MIOpen adds these features. Pull Request resolved: https://github.com/pytorch/pytorch/pull/82002 Approved by: https://github.com/ngimel, https://github.com/malfet
Author
Committer
Parents
Loading