llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib
History
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
..
CAPI Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
Conversion [MLIR][TORCH] Fix OnnxToLinalg lowering issue for ReduceMean op (#3008) 2024-04-02 16:54:04 +05:30
Dialect [ONNX] Preliminary Work Towards Supporting QuantizedMLP_basic onnx e2e test (#3089) 2024-04-01 16:21:05 -07:00
RefBackend [TorchToLinalg] Lower aten.cat to tensor.concat (#2650) 2023-12-15 15:45:32 -05:00
CMakeLists.txt [Stablehlo] lowering chlo to stablehlo in torch-to-stablehlo pipeline (#3037) 2024-03-19 21:18:54 +08:00
InitAll.cpp [Stablehlo] lowering chlo to stablehlo in torch-to-stablehlo pipeline (#3037) 2024-03-19 21:18:54 +08:00