torch-mlir/python/torch_mlir
Ze Zhang 7b73e0cfaf
Add e2e linalg support for aten.atan (#2070)
* new atan op

* update shape

---------

Co-authored-by: Ze Zhang <ze.zhang@getcruise.com>
2023-04-28 00:04:58 -07:00
..
_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 e2e linalg support for aten.atan (#2070) 2023-04-28 00:04:58 -07:00
__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