torch-mlir/lib/Conversion/TorchToLinalg
kumardeepakamd 9adad9bc40
Add support for reflection_pad1d (#2706)
Adds a lowering to Linalg for reflection_pad1d. Based on ideas/code from draft PR
https://github.com/llvm/torch-mlir/pull/2693.

---------

Co-authored-by: Kumar Deepak <kumar@xilinx.com>
2024-01-02 14:05:11 -05:00
..
CMakeLists.txt Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
DataMovement.cpp Add support for reflection_pad1d (#2706) 2024-01-02 14:05:11 -05:00
IndirectDataMovement.cpp [TorchToLinalg] NFC: Move Utils.h to an externally accessible location (#2603) 2023-12-01 19:38:21 -05:00
Linear.cpp [aten] Make `torch.aten.matmul` to `linalg` work for non-broadcasting case (#2659) 2023-12-20 10:09:10 -08:00
Pooling.cpp [TorchToLinalg] NFC: Move Utils.h to an externally accessible location (#2603) 2023-12-01 19:38:21 -05:00
PopulatePatterns.h Re-enable custom op support 2022-08-16 22:49:08 +05:30
Random.cpp [TorchToLinalg] NFC: Move Utils.h to an externally accessible location (#2603) 2023-12-01 19:38:21 -05:00
Reduction.cpp [TorchToLinalg] Drop constexpr from ifs in argmin/max.dim (#2617) 2023-12-07 13:08:17 -05:00
TensorConstructors.cpp [TorchToLinalg] NFC: Move Utils.h to an externally accessible location (#2603) 2023-12-01 19:38:21 -05:00
TensorScalarInterop.cpp Elide dynamic broadcast checks when in strict symbolic shapes mode. (#2496) 2023-09-29 16:45:48 -07:00
TorchToLinalg.cpp Add complex types support with basic complex ops. 2023-05-11 21:29:07 +05:30
Uncategorized.cpp lower torch.aten.isinf to linalg (#2638) 2023-12-28 17:20:32 -08:00
Utils.cpp [TorchToLinalg] NFC: Move Utils.h to an externally accessible location (#2603) 2023-12-01 19:38:21 -05:00