torch-mlir/lib/Dialect/Torch
Vivek Khandelwal 8da7d90611 [MLIR][TORCH] Add E2E support for aten.index_put op
This commit decomposes `aten.index_put` op into
`valsem.aten.index_put_impl` op.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-03-16 22:02:02 +05:30
..
IR Elide `!torch.` prefix in nested dialect types. 2022-03-15 17:24:08 -07:00
Transforms [MLIR][TORCH] Add E2E support for aten.index_put op 2022-03-16 22:02:02 +05:30
Utils Avoid `using` the `torch_upstream` namespace. 2022-03-15 17:24:17 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00