torch-mlir/lib/Dialect/Torch
Vivek Khandelwal d81747eadb
[MLIR][TORCH] Extend support for OnnxToLinalg lowering for Dropout and Div op (#2938)
Fixes https://github.com/nod-ai/SHARK-Turbine/issues/451,
https://github.com/nod-ai/SHARK-Turbine/issues/452
2024-02-27 11:02:05 +05:30
..
IR [MLIR][TORCH] Extend support for OnnxToLinalg lowering for Dropout and Div op (#2938) 2024-02-27 11:02:05 +05:30
Transforms Implement lowering of torch.aten.norm.Scalar (#2899) 2024-02-26 08:46:56 -08:00
Utils Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00