torch-mlir/python/torch_mlir
Vivek Khandelwal 82fb9c7fb8 [MLIR][TORCH] Add decomposition for prims::squeeze op
This commit adds the decomposition for the prims.squeeze op.

Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2023-04-01 21:45:58 +05:30
..
_torch_mlir_custom_op_example [custom op] Generalize shape library logic to work with dtypes (#1594) 2022-12-13 08:25:41 -08:00
cmake/modules Add initial LTC backend (#610) 2022-07-30 09:40:02 -04:00
csrc Update LLVM tag to 21f4b84c (#1918) 2023-03-06 22:53:26 -08:00
dialects [MLIR][TORCH] Add decomposition for prims::squeeze op 2023-04-01 21:45:58 +05:30
__init__.py Add `extra_library` kwarg to `torch_mlir.compile` (#1986) 2023-03-30 09:20:19 -07:00
compiler_utils.py Update LLVM tag to 21f4b84c (#1918) 2023-03-06 22:53:26 -08:00
dynamo.py build: manually update PyTorch version 2023-02-23 11:33:41 +05:30