torch-mlir/lib/Conversion/TorchToLinalg
Rob Suderman f6f890520b
[torch][quant] Quantized `torch.mm` for linalg with end-to-end test (#2750)
This includes custom op matching for decomposed operations and fusing
dequantization into dense operations. As a validation we compare
to the dequant+mm torch implementation.
2024-01-24 14:02:50 -08:00
..
CMakeLists.txt Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
DataMovement.cpp [TorchToLinalg] Add lowering for torch.aten.diagonal (#2632) 2024-01-22 12:47:13 -05:00
IndirectDataMovement.cpp [TorchToLinalg] NFC: Move Utils.h to an externally accessible location (#2603) 2023-12-01 19:38:21 -05:00
Linear.cpp [torch][quant] Quantized `torch.mm` for linalg with end-to-end test (#2750) 2024-01-24 14:02:50 -08:00
Pooling.cpp AtenAdaptiveMaxPool2d Conversion to Linalg (#2779) 2024-01-24 09:09:56 -08:00
PopulatePatterns.h Re-enable custom op support 2022-08-16 22:49:08 +05:30
Random.cpp [TorchToLinalg] NFC: Move Utils.h to an externally accessible location (#2603) 2023-12-01 19:38:21 -05:00
Reduction.cpp [TorchToLinalg] Drop constexpr from ifs in argmin/max.dim (#2617) 2023-12-07 13:08:17 -05:00
TensorConstructors.cpp Fix unused variable warnings (#2775) 2024-01-22 11:05:55 -08:00
TensorScalarInterop.cpp Elide dynamic broadcast checks when in strict symbolic shapes mode. (#2496) 2023-09-29 16:45:48 -07:00
TorchToLinalg.cpp Add complex types support with basic complex ops. 2023-05-11 21:29:07 +05:30
Uncategorized.cpp Implement lowering of torch.aten.remainder.Tensor (#2763) 2024-01-19 18:09:08 +05:30
Utils.cpp [torch][quant] Support quantize and dequantize for torch (#2731) 2024-01-12 19:11:14 -08:00