mirror of https://github.com/llvm/torch-mlir
d8ba68119e
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. |
||
---|---|---|
.. | ||
torchscript |