Rob Suderman
|
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 |
Rob Suderman
|
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 |
Rob Suderman
|
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 |