torch-mlir/lib/Conversion
Gaurav Shukla 3b85c70748
[ONNX][MLIR] Add support for onnx.gather op (#2726)
This commit adds support for gather op in the onnx pipeline.
https://github.com/nod-ai/SHARK-Turbine/issues/242

Signed-off-by: Gaurav Shukla <gaurav.shukla@amd.com>
2024-01-19 21:58:29 +05:30
..
TorchConversionToMLProgram Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
TorchOnnxToTorch [ONNX][MLIR] Add support for onnx.gather op (#2726) 2024-01-19 21:58:29 +05:30
TorchToArith Fix for unused variable failure when trying to bump torch-mlir in IREE (#2560) 2023-11-08 15:55:41 -08:00
TorchToLinalg Add aten.pool_max3d support to torch-to-linalg (#2735) 2024-01-19 21:09:46 +05:30
TorchToSCF Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
TorchToStablehlo Advance llvm-project and stablehlo. (#2619) 2023-12-07 23:13:42 -08:00
TorchToTMTensor Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
TorchToTensor [onnx] Lowering for `onnx.shape` to `torch` and `tensor` (#2648) 2023-12-15 11:37:49 -08:00
TorchToTosa torch-to-tosa lowering support for AtenLinalgVectorNormOp (#2734) 2024-01-18 12:32:23 -08:00
Utils [Torch Dialect]Support aten.cosine_similarity (#2364) 2023-11-08 15:28:30 +08:00
CMakeLists.txt [onnx] Lowering for `onnx.shape` to `torch` and `tensor` (#2648) 2023-12-15 11:37:49 -08:00
PassDetail.h Minor fixes for `ConvertTorchConversionToMLProgram`. (#1991) 2023-04-04 09:09:58 -07:00
Passes.cpp [onnx] Lowering for `onnx.shape` to `torch` and `tensor` (#2648) 2023-12-15 11:37:49 -08:00