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-01-15 23:23:11 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
f8aeecef46461ff574f51982d03310fa8c57888e
ColossalAI/colossalai/nn/_ops
History
jim e5ab6be72e [hotfix[ fix colotensor.type() raise NotImplementedError (#1682)
2022-10-10 10:13:31 +08:00
..
__init__.py
[FAW] export FAW in _ops (#1438)
2022-08-11 13:43:24 +08:00
_utils.py
[embedding] tablewise sharding polish (#1535)
2022-09-02 11:09:37 +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
[hotfix[ fix colotensor.type() raise NotImplementedError (#1682)
2022-10-10 10:13:31 +08:00
embedding_bag.py
[NFC] polish colossalai/nn/_ops/embedding_bag.py code style (#1552)
2022-09-08 22:11:04 +08:00
embedding.py
[NFC] polish colossalai/nn/_ops/embedding.py code style (#1561)
2022-09-08 22:11:04 +08:00
layernorm.py
[NFC] polish colossalai/nn/_ops/layernorm.py code style (#1555)
2022-09-08 22:11:04 +08:00
linear.py
[tensor] added linear implementation for the new sharding spec (#1416)
2022-08-12 11:33:09 +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: 1504ms Template: 28ms
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