torch-mlir/lib/Dialect/Torch
zjgarvey 99115dcdc8
[Torch] Address unnecessary dynamic shapes in argmax decomposition (#3889)
Addresses <https://github.com/iree-org/iree/issues/19262#issue>
2024-11-22 16:03:29 -08:00
..
IR [Torch Dialect] Add `torch.aten.mul.int_float` (required to simplify shape calculation of `upsample_nearest2d`) (#3764) 2024-11-21 00:43:06 +08:00
Transforms [Torch] Address unnecessary dynamic shapes in argmax decomposition (#3889) 2024-11-22 16:03:29 -08:00
Utils Add Op for `torch.aten.unfold` (#3772) 2024-10-08 21:10:43 +00:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00