Commit Graph

4 Commits

Author SHA1 Message Date
Jiarui Fang
e76f76c08b
[Tensor] test parameters() as member function (#896) 2022-04-28 10:57:14 +08:00
Ziyue Jiang
cb182da7c5
[tensor] refine linear and add gather for laynorm (#893)
* refine linear and add function to ColoTensor

* add gather for layernorm

* polish

* polish
2022-04-28 10:55:40 +08:00
Jiarui Fang
26c49639d8
[Tensor] overriding paramters() for Module using ColoTensor (#889) 2022-04-27 15:28:59 +08:00
Jiarui Fang
a0e5971692
[Tensor] test model check results for a simple net (#887) 2022-04-27 12:00:18 +08:00