torch-mlir

History

Yi Zhang d8ba68119e Lower aten::view with linalg.collapse and linalg.expand We only handle the expanding OR collapsing cases, we do not handle expanding And collapsing happening at the same time or cases where it's neither collapsing nor expanding like view of [2,3] for 3x2 tensor. It's assumed that if a shape list element is got from `aten.size(tensor, dim)` the corresponding dim is not splitted or collapsed. This assumption makes it easier to deal with dynamic shapes.	2021-12-16 17:58:20 -05:00
..
CMakeLists.txt	Move external/torch-mlir to the root of the repo.	2021-09-27 17:11:08 -07:00
TorchToLinalg.cpp	Lower aten::view with linalg.collapse and linalg.expand	2021-12-16 17:58:20 -05:00

Yi Zhang d8ba68119e Lower aten::view with linalg.collapse and linalg.expand

We only handle the expanding OR collapsing cases, we do not handle
expanding And collapsing happening at the same time or cases where
it's neither collapsing nor expanding like view of [2,3] for
3x2 tensor.

It's assumed that if a shape list element is got from
`aten.size(tensor, dim)` the corresponding dim is not splitted or
collapsed. This assumption makes it easier to deal with dynamic shapes.

2021-12-16 17:58:20 -05:00

CMakeLists.txt

Move external/torch-mlir to the root of the repo.

2021-09-27 17:11:08 -07:00

TorchToLinalg.cpp

Lower aten::view with linalg.collapse and linalg.expand

2021-12-16 17:58:20 -05:00