torch-mlir/lib/Conversion
jinchen 032a636c35
Fix onnx.If lowering with scalar condition tensor (#3846)
Fixes
https://github.com/nod-ai/SHARK-ModelDev/issues/696#issuecomment-2442016530
2024-10-31 20:34:50 -07:00
..
TorchConversionToMLProgram [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
TorchOnnxToTorch Fix onnx.If lowering with scalar condition tensor (#3846) 2024-10-31 20:34:50 -07:00
TorchToArith [TorchToArith] Add a lowering for `torch.add.float_int` (#3594) 2024-08-07 11:55:27 -05:00
TorchToLinalg [linalg] Fix torch.aten.add of `torch.bool` (#3820) 2024-10-31 17:37:25 -07:00
TorchToSCF [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
TorchToStablehlo [Torch] support AtenExp2Op (#3832) 2024-10-31 19:14:05 +08:00
TorchToTMTensor [linalg] Broadcast batch for mask on sdpa lowering (#3824) 2024-10-31 17:59:24 -07:00
TorchToTensor [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
TorchToTosa [TOSA] Expand Torch to TOSA legalization coverage (#3827) 2024-10-30 16:26:10 -07:00
Utils [LINALG] Add complex tensor support for `create[Zero|One]InitTensor` utility (#3777) 2024-10-09 16:15:08 +05:30
CMakeLists.txt [torch] Improve shape inference for `torch-to-linalg` path for reshapes (#3055) 2024-03-26 12:41:40 -07:00
PassDetail.h Minor fixes for `ConvertTorchConversionToMLProgram`. (#1991) 2023-04-04 09:09:58 -07:00
Passes.cpp Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00