Logo
Explore Help
Register Sign In
github/ColossalAI
1
0
Fork 0
You've already forked ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-04 09:52:50 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
f20cb4e893cedf1d78e753fbac557232af06720b
ColossalAI/colossalai/nn/_ops
History
Jiarui Fang 10b3df65c8 [FAW] move coloparam setting in test code. (#1429)
2022-08-10 14:31:53 +08:00
..
cache_embedding
[FAW] move coloparam setting in test code. (#1429)
2022-08-10 14:31:53 +08:00
__init__.py
[colotensor] add Tensor.view op and its unit test (#1343)
2022-07-21 10:53:15 +08:00
_utils.py
[FAW] parallel FreqAwareEmbedding (#1424)
2022-08-10 13:44:30 +08:00
addmm.py
[colotensor] add Tensor.view op and its unit test (#1343)
2022-07-21 10:53:15 +08:00
element_wise.py
[tensor] add zero_like colo op, important for Optimizer (#1236)
2022-07-08 14:55:27 +08:00
embedding_bag.py
[colotensor] add Tensor.view op and its unit test (#1343)
2022-07-21 10:53:15 +08:00
embedding.py
[colotensor] add Tensor.view op and its unit test (#1343)
2022-07-21 10:53:15 +08:00
layernorm.py
[colotensor] add Tensor.view op and its unit test (#1343)
2022-07-21 10:53:15 +08:00
linear.py
[hotfix] fix shape error in backward when using ColoTensor (#1298)
2022-07-13 23:06:12 +08:00
loss.py
[tensor] fix a assertion in colo_tensor cross_entropy (#1232)
2022-07-08 11:18:00 +08:00
view.py
[colotensor] add Tensor.view op and its unit test (#1343)
2022-07-21 10:53:15 +08:00
Powered by Gitea Version: 1.25.2 Page: 857ms Template: 50ms
English
Bahasa Indonesia Deutsch English Español Français Gaeilge Italiano Latviešu Magyar nyelv Nederlands Polski Português de Portugal Português do Brasil Suomi Svenska Türkçe Čeština Ελληνικά Български Русский Українська فارسی മലയാളം 日本語 简体中文 繁體中文(台灣) 繁體中文(香港) 한국어
Licenses API