Jiarui Fang 845856ea29 [Graph] building computing graph with ColoTensor, Linear only (#917)

2022-05-07 17:10:37 +08:00

..

__init__.py

[Tensor] add ColoTensor TP1Dcol Embedding (#899)

2022-04-28 17:45:06 +08:00

element_wise.py

[Tensor] make ColoTensor more robust for getattr (#886)

2022-04-27 10:57:49 +08:00

embedding.py

[Tensor] add embedding tp1d row (#904)

2022-04-29 14:10:05 +08:00

layernorm.py

[Tensor] add ColoTensor TP1Dcol Embedding (#899)

2022-04-28 17:45:06 +08:00

linear.py

[Graph] building computing graph with ColoTensor, Linear only (#917)

2022-05-07 17:10:37 +08:00

loss.py

[Tensor] add 1d vocab loss (#918)

2022-05-07 15:49:14 +08:00