Skip to content

Commit

Permalink
[Documentation] - Annotate attention kernel with shapes of tensors (#…
Browse files Browse the repository at this point in the history 
…1244)

* add attention shape annotations

* linting

* fix
  • Loading branch information
@goliaro
goliaro authored Dec 12, 2023
1 parent 08f60b1 commit 3cf49a6
Show file tree
Hide file tree
Showing 2 changed files with 247 additions and 198 deletions.
Loading

0 comments on commit 3cf49a6

Please sign in to comment.