torch-mlir/lib/Dialect/Torch
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
..
IR Remove folder from `AtenStackOp` for single element list inputs (#2626) 2023-12-11 10:52:50 -08:00
Transforms Add support for reflection_pad1d (#2706) 2024-01-02 14:05:11 -05:00
Utils [onnx] Lower onnx.HardSigmoid to torch (#2682) 2023-12-21 07:29:22 -08:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00