torch-mlir/test/Conversion/TorchToLinalg
Aart Bik 4361178caa
[torch-mlir][sparse] recognize sparse tensor conversion (#3226)
Sparse tensor conversions are represented by special aten operators.
This PR ensures the conversions are recognized (instead of failing the
full torch aten lowering to linalg).
2024-04-26 02:32:07 +08:00
..
basic.mlir Drop torch attributes at the end of backend conversion. (#2876) 2024-02-13 14:32:02 -08:00
broadcast.mlir [TorchToLinalg] Improve broadcast lowerings in strict symbolic modes (#2505) 2023-10-05 15:15:26 -04:00
convolution.mlir [MLIR][Torch] Do not convert bias tensor to element type if NoneType (#3072) 2024-04-02 14:19:26 +05:30
elementwise.mlir [torch] Lower `torch.aten.sinh` to `linalg` (#2662) 2023-12-18 09:15:12 -08:00
flatten.mlir Implement Expand/Collapse Functionality for Aten.View (#1353) 2022-09-27 11:08:14 -07:00
gridsampler.mlir [onnx][torch][linalg] Implementing align-corner modes for gridsampler (#3171) 2024-04-17 13:38:19 -07:00
pooling.mlir [torch] GridSample TorchToLinalg lowering (#2883) 2024-02-23 09:14:38 -08:00
sparse.mlir [torch-mlir][sparse] recognize sparse tensor conversion (#3226) 2024-04-26 02:32:07 +08:00
unsqueeze.mlir mlir: bump llvm tag to 5380e3 (#856) 2022-05-16 12:54:35 -07:00
view.mlir [torch] Add support for `torch.view` with dynamic shapes (#3164) 2024-04-18 11:47:19 -07:00