Skip to content

Commit

Permalink
Allow expanding and collapsing in aten::view
Browse files Browse the repository at this point in the history 
 - Supports cases where the view op expands and collapses dims
simulataneously. This does not handle the case where it is neither
expanding nor collapsing (e.g. [2, 3] -> [3, 2])
 - Additionally fixes a previous bug with adding 1-sized dims on both
sides of a tensor with aten.view
  • Loading branch information
@qedawkins
qedawkins committed Jul 19, 2022
1 parent 7f08169 commit 452905f
Show file tree
Hide file tree
Showing 2 changed files with 344 additions and 144 deletions.
Loading

0 comments on commit 452905f

Please sign in to comment.