github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 09:42:27 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
32291dd73f92843b4f221226a95f2f7048b1766b
ColossalAI/colossalai/tensor/modules
History
Ziyue Jiang 32291dd73f [Tensor] add module handler for linear (#1021) 
* add module spec for linear

* polish

* polish

* polish
2022-05-26 11:50:44 +08:00
..
__init__.py
[Tensor] add module handler for linear (#1021)
2022-05-26 11:50:44 +08:00
colo_module.py
[Tensor] add module handler for linear (#1021)
2022-05-26 11:50:44 +08:00
linear.py
[Tensor] add module handler for linear (#1021)
2022-05-26 11:50:44 +08:00