Skip to content

Commit

Permalink
[Relax][PyTorch] Cleanup Tensor Manipulation and Creation op converte…
Browse files Browse the repository at this point in the history
…rs (#17376)

* cleanup `_cat()`

* cleanup `_cumsum()`

* cleanup `_expand()`

* cleanup `_flatten()`

* cleanup `_permute()`

* cleanup `_repeat()`

* cleanup `_reshape()`

* cleanup `_size()`

* cleanup `_split()`

* cleanup `_squeeze()`

* cleanup `_tile()`

* cleanup `_transpose()`

* cleanup `chunk()`

* cleanup `_arange()`

* cleanup `_empty()`

* cleanup `_inplace_fill()`

* cleanup `_full()`

* cleanup `_index_select()`

* cleanup `_inplace_masked_fill()`

* cleanup `_masked_fill()`

* cleanup `_new_ones()`

* cleanup `_ones()`

* cleanup `_tensor()`

* `_inplace_tril_triu()` is an unary op

* `_batch_norm_2d()` is a nn ops

* `_interpolate()` is a nn ops

* `_cross_entropy()` is a nn ops

* chore

* fix tensor size
  • Loading branch information
mshr-h committed Sep 16, 2024
1 parent 11198f6 commit 4ab3f82
Showing 1 changed file with 358 additions and 397 deletions.
Loading

0 comments on commit 4ab3f82

Please sign in to comment.