torch-mlir/lib/Conversion/TorchToTMTensor
Rob Suderman e9cdd6cbc5
[torch] Fix tm_tensor.attention for end-to-end (#2907)
Some operations include a backend matcher for specialized operations. We
map these back to generics so they appropriately match to the high
performance versions. This is done for the attention operation.
2024-02-13 21:18:01 -08:00
..
CMakeLists.txt Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
TorchToTMTensor.cpp [torch] Fix tm_tensor.attention for end-to-end (#2907) 2024-02-13 21:18:01 -08:00