Skip to content

Commit

Permalink
[Relax][PyTorch] Cleanup Neural Network op converters (#17369)
Browse files Browse the repository at this point in the history
* cleanup `_adaptive_avg_pool2d()`

* cleanup `addmm()`

* cleanup `_avg_pool2d()`

* cleanup `_baddbmm()`

* cleanup `_conv1d_transpose()`

* cleanup `_conv2d_transpose()`

* cleanup `_conv1d()`

* cleanup `_conv2d()`

* cleanup `_conv3d()`

* cleanup `_einsum()`

* cleanup `_embedding()`

* cleanup `_group_norm()`

* cleanup `_layer_norm()`

* cleanup `_linear()`

* cleanup `_max_pool2d()`

* cleanup `_scaled_dot_product_attention()`

* cleanup `_unbind()`

* remove `_matmul_impl()` since we don't use it anymore
  • Loading branch information
mshr-h authored Sep 13, 2024
1 parent 751467e commit 3755571
Showing 1 changed file with 755 additions and 771 deletions.
Loading

0 comments on commit 3755571

Please sign in to comment.