Vivek Khandelwal 378860f51b [MLIR][TORCH] Add E2E support for aten.topk op ... This commit adds the decomposition for the aten.topk op. Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>		2023-05-05 15:50:33 +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	Add Shape inference for CopyOp for lazy tensor core backend (#2006)	2023-04-12 09:37:03 -04:00
dialects	[MLIR][TORCH] Add E2E support for aten.topk op	2023-05-05 15:50:33 +05:30
__init__.py	Add `extra_library` kwarg to `torch_mlir.compile` (#1986)	2023-03-30 09:20:19 -07:00
compiler_utils.py	[Dynamo] turn on `no_python=True` for dynamo tests (#2040)	2023-04-28 18:05:17 -05:00
dynamo.py	[TorchDynamo] Add aten.squeeze op to the decomposition list	2023-04-06 22:21:25 +05:30