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