This website requires JavaScript.
Explore
Help
Register
Sign In
github
/
ColossalAI
Watch
1
Star
0
Fork
0
You've already forked ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI.git
synced
2026-01-17 16:08:30 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
d209aff6845cf713531acee68dd8e2d22381cfee
ColossalAI
/
colossalai
/
nn
/
_ops
/
cache_embedding
History
Jiarui Fang
d209aff684
Add FreqAwareEmbeddingBag (
#1421
)
2022-08-09 16:26:12 +08:00
..
__init__.py
Add FreqAwareEmbeddingBag (
#1421
)
2022-08-09 16:26:12 +08:00
base_embedding.py
[FAW] add cache manager for the cached embedding (
#1419
)
2022-08-09 15:17:17 +08:00
cache_mgr.py
[FAW] add cache manager for the cached embedding (
#1419
)
2022-08-09 15:17:17 +08:00
copyer.py
[FAW] add cache manager for the cached embedding (
#1419
)
2022-08-09 15:17:17 +08:00
freq_aware_embedding.py
Add FreqAwareEmbeddingBag (
#1421
)
2022-08-09 16:26:12 +08:00
