torch-mlir/lib/Dialect/Torch
Vivek Khandelwal 5e5d51e4b5 Fold brodcast ops for llama2 2023-10-04 19:17:09 +00:00
..
IR Fold brodcast ops for llama2 2023-10-04 19:17:09 +00:00
Transforms [MLIR][TORCH] Add support for bitwise_right_shit and bitwise_and.Scalar op 2023-10-02 13:06:59 +05:30
Utils Elide dynamic broadcast checks when in strict symbolic shapes mode. (#2496) 2023-09-29 16:45:48 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00