torch-mlir/lib/Dialect/Torch
Yuanqiang Liu 39201a4be5
[Torch Dialect] avoid assertion failure when PrimNumToTensorScalarOp'… (#2256)
* [Torch Dialect] avoid assertion failure when PrimNumToTensorScalarOp's input is torch.number

* update
2023-06-23 16:02:45 +08:00
..
IR [Torch Dialect] Support aten.device.with_index (#2254) 2023-06-23 01:07:14 +08:00
Transforms [Torch Dialect] avoid assertion failure when PrimNumToTensorScalarOp'… (#2256) 2023-06-23 16:02:45 +08:00
Utils [Torch Dialect] fix torch.uint8's dtype infer (#2227) 2023-06-13 10:38:20 +08:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00