torch-mlir/lib/Dialect/Torch
gpetters94 9ec0683e92
Add 2D case for convolution (#693)
2022-04-08 00:47:57 -04:00
..
IR Fix for 0-size dim inferred incorrectly. 2022-03-30 16:36:15 -07:00
Transforms Add 2D case for convolution (#693) 2022-04-08 00:47:57 -04:00
Utils Add support for multi-dim input to `index_put_impl` (#722) 2022-03-31 09:27:21 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00