torch-mlir/python/torch_mlir
rahul shrivastava e3d876af42 Add aten.scatter.value Op ODS
Signed-off-by: rahul shrivastava <rahul.shrivastava@cerebras.net>
2023-04-25 11:40:19 +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 Add aten.scatter.value Op ODS 2023-04-25 11:40:19 +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 [TorchDynamo] Add aten.squeeze op to the decomposition list 2023-04-06 22:21:25 +05:30