torch-mlir/lib/Dialect/Torch
James Newling f78ec78ac8
Adjust bound check to be the same as PyTorch native (i.e. stricter) (#2755)
prims.expand expects the start and end dimensions to be strictly less
than the rank of the tensor.
2024-01-15 11:44:45 -08:00
..
IR [torch][quant] Support quantize and dequantize for torch (#2731) 2024-01-12 19:11:14 -08:00
Transforms Adjust bound check to be the same as PyTorch native (i.e. stricter) (#2755) 2024-01-15 11:44:45 -08:00
Utils [torch][quant] Support quantize and dequantize for torch (#2731) 2024-01-12 19:11:14 -08:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00