torch-mlir/projects/pt1/python/torch_mlir
Vivek Khandelwal d6d1a173dc
[MLIR][Torch] Add OnnxToTorch and TorchToLinalg support for trig ops (#2903)
This commit adds the OnnxToTorch lowering for cosh, acosh, asin, asinh,
and atanh op.
This commit also adds the TorchToLinalg lowering for acosh, asin, asinh,
and atanh op.

Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-02-14 11:58:09 +05:30
..
_torch_mlir_custom_op_example Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
csrc Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
jit_ir_importer [MLIR][Torch] Add OnnxToTorch and TorchToLinalg support for trig ops (#2903) 2024-02-14 11:58:09 +05:30
_dynamo_fx_importer.py Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
_version.py Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
compiler_utils.py Allow printing all IR in `torch_mlir.compile` (#2669) 2023-12-20 15:08:21 -06:00
dynamo.py Rename torch_mlir.compile APIs and introduce FX based analogs (#2842) 2024-02-06 19:07:59 -08:00
torchscript.py Rename torch_mlir.compile APIs and introduce FX based analogs (#2842) 2024-02-06 19:07:59 -08:00