Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement EinsumDense layer #7860

Merged
merged 16 commits into from
Jul 26, 2023
Merged

Implement EinsumDense layer #7860

Show file tree
Hide file tree
Changes from 13 commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
ea472e2
Add spec for multi-head attention
pforderique Jul 13, 2023
20f8358
Merge branch 'main' into spec-transformer
pforderique Jul 13, 2023
41a105e
Add CachedMultiHeadAttention cache
pforderique Jul 13, 2023
6e78ffc
Fix typos
pforderique Jul 13, 2023
01d9e2e
Lint
pforderique Jul 13, 2023
8f08c19
Add Transformer Decoder spec
pforderique Jul 13, 2023
4713c4e
lint
pforderique Jul 14, 2023
37aca1a
Add Einsum spec
pforderique Jul 14, 2023
2a6d929
lint
pforderique Jul 14, 2023
6dcb7a0
Remove unused type declaration
pforderique Jul 17, 2023
db6fc8d
Merge branch 'main' into spec-transformer
pforderique Jul 17, 2023
e589817
Move helper functions outside EinsumDense class
pforderique Jul 17, 2023
9bafba5
Implement Einsum Dense
pforderique Jul 19, 2023
4428cf1
Address comments
pforderique Jul 26, 2023
871dc34
Merge branch 'main' into einsum-dense-impl
pforderique Jul 26, 2023
ec52b22
Merge branch 'master' into einsum-dense-impl
pforderique Jul 26, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading