14b548f968
[torch] Improve shape inference for `torch-to-linalg` path for reshapes ( #3055 )
...
Reshaping tensors depend on directly matching individual dimensions to
their corresponding dim in the `torch.view` reshape dimensions. This
involves decoupling dynamic dimensions from their static counterparts
and support cleanup / canonicalization.
2024-03-26 12:41:40 -07:00
3a56714bff
[torch] Fix clamp ranges on quantize_per_tensor on unsigned ( #3018 )
...
SExtValue was used for `int` and `uint` clamp values. This caused the
result to always be outputed as `zero`.
2024-03-20 13:37:47 -07:00
dd673cfa8d
[torch] Add edgecase for aten.shape_to_tensor for rank-0 input ( #2962 )
...
Currently lowering uses `tensor.from_elements` which does not allow zero
inputs. In this case we return a `tensor.empty` operation.
2024-02-28 09:47:06 -08:00
0114a570e3
[torch] Support lowering `torch.item` to `tensor.extract` ( #2835 )
...
Extracting scalar values from tensors can be implemented via a lowering
to tensor.extract.
2024-01-31 15:09:12 -08:00
061af696ce
[onnx] Lowering for `onnx.shape` to `torch` and `tensor` ( #2648 )
...
Includes the lowering from the `aten` equivalent to `tensor` operations.
2023-12-15 11:37:49 -08:00
Rob Suderman