Ziyue Jiang 797a9dc5a9 add DistSpec for loss and test_model (#947)		2022-05-13 20:29:50 +08:00
..
__init__.py	add DistSpec for loss and test_model (#947)	2022-05-13 20:29:50 +08:00
addmm.py	[tensor] design DistSpec and DistSpecManager for ColoTensor (#934)	2022-05-13 15:13:52 +08:00
element_wise.py	[Tensor] make ColoTensor more robust for getattr (#886)	2022-04-27 10:57:49 +08:00
embedding.py	[tensor] design DistSpec and DistSpecManager for ColoTensor (#934)	2022-05-13 15:13:52 +08:00
layernorm.py	add DistSpec for loss and test_model (#947)	2022-05-13 20:29:50 +08:00
linear.py	[tensor] design DistSpec and DistSpecManager for ColoTensor (#934)	2022-05-13 15:13:52 +08:00
loss.py	add DistSpec for loss and test_model (#947)	2022-05-13 20:29:50 +08:00