torch-mlir/lib/Dialect
zjgarvey e346c911f7
[ONNX] Add basic support for RoiAlign (#3493)
This adds an onnx->torch conversion for onnx.RoiAlign into
torchvision.roi_align or torchvision.roi_pool, and adds those two
torchvision ops to torch-mlir.
2024-06-25 11:02:45 -05:00
..
TMTensor Implement lowering of torch.aten.kthvalue (#3360) 2024-06-15 11:18:39 +05:30
Torch [ONNX] Add basic support for RoiAlign (#3493) 2024-06-25 11:02:45 -05:00
TorchConversion [Torch] fix toBuiltinTensor() (#3415) 2024-06-08 09:36:32 +08:00
CMakeLists.txt Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00