torch-mlir/lib/Conversion
Rob Suderman c15f1a2bd2
[onnx] Adding lowering for `onnx.Size` operation (#2985)
We can support `onnx.Size` by requesing the size of each dimensions and
taking the product of the results, then packing it into a tensor.

---------

Co-authored-by: Scott Todd <scott.todd0@gmail.com>
2024-03-06 17:01:05 -08:00
..
TorchConversionToMLProgram Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
TorchOnnxToTorch [onnx] Adding lowering for `onnx.Size` operation (#2985) 2024-03-06 17:01:05 -08:00
TorchToArith [torch] Supporting `torch.aten.mul.float` lowering to `arith` (#2833) 2024-02-05 16:23:04 -08:00
TorchToLinalg [onnx] Migrate `onnx.ReduceMax` to match `onnx.ReduceMin` (#2981) 2024-03-06 16:48:21 -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 torch_to_stablehlo::getBackendTypeForScalarType (#2975) 2024-03-04 23:31:54 +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 [MLIR][TORCH] Fix OnnxToLinalg lowering issue for sub and sum op (#2954) 2024-02-29 21:48:46 +05:30
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