torch-mlir/lib/Dialect/Torch
Vivek Khandelwal 661be2d5b0
[MLIR][Torch] Add TorchToLinalg lowering for AtenAvgPool3dOp (#3030)
This commit also fixes the average pool op' test failing for
OnnxToLinalg lowering.

Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-06-04 22:12:34 +05:30
..
IR [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
Transforms [MLIR][Torch] Add TorchToLinalg lowering for AtenAvgPool3dOp (#3030) 2024-06-04 22:12:34 +05:30
Utils [ONNX] Add OnnxToTorch lowering for SpaceToDepth op (#3393) 2024-06-03 20:29:39 +05:30
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00