torch-mlir/lib/Dialect/Torch
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
..
IR Implement lowering of torch.aten.triu_indices (#3451) 2024-06-21 16:16:38 -07:00
Transforms [ONNX] Add basic support for RoiAlign (#3493) 2024-06-25 11:02:45 -05:00
Utils [TorchToLinalg] Fix Quantized Convolution Accumulator Type (#3459) 2024-06-20 13:54:20 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00