Skip to content

Commit

Permalink
[Relax][PyTorch] Support neural network ops for ExportedProgram impor…
Browse files Browse the repository at this point in the history
…ter (#17426)

* support batchnorm2d and getitem

* support addmm

* support avg_pool2d

* support baddbmm

* support bmm

* support conv_transpose1d

* support conv_transpose2d

* support conv1d

* support conv3d

* support einsum

* support embedding

* support group_norm

* support layer_norm

* support scaled_dot_product_attention

* support unbind

* support interpolate

* fix lint error
  • Loading branch information
mshr-h authored Sep 28, 2024
1 parent 7ff4d0d commit d9ee637
Show file tree
Hide file tree
Showing 4 changed files with 1,723 additions and 484 deletions.
Loading

0 comments on commit d9ee637

Please sign in to comment.