6 Commits

Author	SHA1	Message	Date
Jiarui Fang	7f76517a85	[Tensor] make a simple net works with 1D row TP (#879)	2022-04-26 18:11:47 +08:00
Jiarui Fang	e43f83aa5c	[Tensor] get named parameters for model using ColoTensors (#874)	2022-04-26 14:08:01 +08:00
Jiarui Fang	96211c2cc8	[tensor] customized op returns ColoTensor (#875) ... * [tensor] customized op returns ColoTensor * polish * polish code	2022-04-26 13:23:59 +08:00
Jiarui Fang	1190b2c4a4	[tensor] add cross_entrophy_loss (#868)	2022-04-25 16:01:52 +08:00
Ziyue Jiang	bcc8655021	[Tensor ] Add 1Drow weight reshard by spec (#854)	2022-04-24 18:30:20 +08:00
Jiarui Fang	62f059251b	[Tensor] init a tp network training unittest (#849)	2022-04-24 16:43:44 +08:00