torch-mlir/lib/Dialect/Torch
Yuanqiang Liu e2fbded49c
[Torch Dialect] improve argmax/argmin's decomposition to support keep… (#3514)
…dim=True when dim=None
2024-07-02 09:08:57 +08:00
..
IR [Torch] Fix torch.constant.int operation parsing (#3476) 2024-06-28 16:06:52 +02:00
Transforms [Torch Dialect] improve argmax/argmin's decomposition to support keep… (#3514) 2024-07-02 09:08:57 +08:00
Utils [TorchToLinalg] Fix Quantized Convolution Accumulator Type (#3459) 2024-06-20 13:54:20 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00