Ziyue Jiang f593a5637e [Tensor] add embedding tp1d row (#904)

2022-04-29 14:10:05 +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

[Tensor] add ColoTensor TP1Dcol Embedding (#899)

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

loss.py

[tensor] add cross_entrophy_loss (#868)

2022-04-25 16:01:52 +08:00