torch-mlir/lib/Conversion
Rob Suderman e30a083aff
[torch] Rework lowering to tm_tensor.scatter to stop serialization (#2940)
We collapsed and broadcasted scatter indices to a single element
version. We should instead upport `tm_tensor.scatter`s support for
multiple indices and the implicitly broadcasted behavior. This avoids
the serialization and materializing a needlessly large indices tensor.
2024-02-27 11:46:57 -08:00
..
TorchConversionToMLProgram Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
TorchOnnxToTorch [MLIR][TORCH] Extend support for OnnxToLinalg lowering for Dropout and Div op (#2938) 2024-02-27 11:02:05 +05:30
TorchToArith [torch] Supporting `torch.aten.mul.float` lowering to `arith` (#2833) 2024-02-05 16:23:04 -08:00
TorchToLinalg [MLIR][TORCH] Add support for tanh approximation for Gelu op (#2941) 2024-02-27 19:26:01 +05:30
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] Support lowering `torch.item` to `tensor.extract` (#2835) 2024-01-31 15:09:12 -08:00
TorchToTosa allow tosa.cast to convert from f32 to f16 (#2934) 2024-02-20 14:22:38 -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