torch-mlir/lib/Conversion
Andreas Falkenberg 5437f32193
[onnx][torch] Lower `onnx.grid_sampler` to the `torch` equivalents (#2952)
This is the lowering of gridsampler from onnx to torch using our prior
implementation of AtenGridSamplerOp.
Here are several checks for cornercases implemented. We may decide to
have part of these checks in AtenGridSamplerOp instead of the onnx
lowering portion.
2024-02-28 13:52:15 -08:00
..
TorchConversionToMLProgram Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
TorchOnnxToTorch [onnx][torch] Lower `onnx.grid_sampler` to the `torch` equivalents (#2952) 2024-02-28 13:52:15 -08:00
TorchToArith [torch] Supporting `torch.aten.mul.float` lowering to `arith` (#2833) 2024-02-05 16:23:04 -08:00
TorchToLinalg [torch] Fix folders and `cat` and `view` torch lowerings (#2963) 2024-02-28 12:04:52 -08:00
TorchToSCF Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
TorchToStablehlo [Stablehlo] add refbackend (#2712) 2024-02-16 01:08:48 +08:00
TorchToTMTensor [torch] Rework lowering to tm_tensor.scatter to stop serialization (#2940) 2024-02-27 11:46:57 -08:00
TorchToTensor [torch] Add edgecase for aten.shape_to_tensor for rank-0 input (#2962) 2024-02-28 09:47:06 -08:00
TorchToTosa [tosa] Fix TOSA batch matmul lowering to correct transpose ordering (#2959) 2024-02-28 09:46:58 -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 Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00