torch-mlir/test/Conversion
Vivek Khandelwal d1f770c620
[MLIR][TORCH] Fix OnnxToLinalg lowering issue for ReduceMean op (#3008)
This commit also cleans up the OnnxToTorch lowering for the ReduceMean
op and adds the support for handling edge cases.

Signed-Off By: Vivek Khandelwal vivekkhandelwal1424@gmail.com
2024-04-02 16:54:04 +05:30
..
TorchConversionToMLProgram Bump LLVM and StableHLO (#2598) 2023-11-28 22:12:24 -08:00
TorchOnnxToTorch [MLIR][TORCH] Fix OnnxToLinalg lowering issue for ReduceMean op (#3008) 2024-04-02 16:54:04 +05:30
TorchToArith [MLIR][TORCH]Add support lowing aten.Int.bool to arith (#3083) 2024-04-01 10:05:08 -07:00
TorchToLinalg [MLIR][Torch] Do not convert bias tensor to element type if NoneType (#3072) 2024-04-02 14:19:26 +05:30
TorchToSCF Fix SCF Forloop fails to convert to linalg when a tensor argument is supplied to the loop block (#3040) 2024-03-20 11:04:02 -07:00
TorchToStablehlo [torch] Add folder for `prim.NumToTensor.Scalar` (#2921) 2024-02-19 11:55:54 -08:00
TorchToTensor [onnx] Lowering for `onnx.shape` to `torch` and `tensor` (#2648) 2023-12-15 11:37:49 -08:00
TorchToTosa allow tosa.cast to convert from f32 to f16 (#2934) 2024-02-20 14:22:38 -08:00