summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/transformer/embeddings/axial.py
AgeCommit message (Collapse)Author
2023-08-25Rename and add flash attenGustaf Rydholm
2022-10-02Add commentsGustaf Rydholm
2022-09-18Format importsGustaf Rydholm
Format imports
2022-09-05Steal lucidrains axial encodingGustaf Rydholm
2021-10-27Add axial embeddingGustaf Rydholm