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