pytorch
85698d0a - add conv_transpose2d pointwise(unary) fusion kernel (#90264)

Commit
2 years ago
add conv_transpose2d pointwise(unary) fusion kernel (#90264) This PR adds `torch.ops.mkldnn._convolution_transpose_pointwise` that supports ConvTranspose fusion with the below unary pointwise OPs: - relu - sigmoid - tanh - hardswish - leaky_relu - hardtanh - gelu Pull Request resolved: https://github.com/pytorch/pytorch/pull/90264 Approved by: https://github.com/jgong5, https://github.com/jansel
Author
Committer
Parents
Loading