Ziyue Jiang cb182da7c5 [tensor] refine linear and add gather for laynorm (#893) ... * refine linear and add function to ColoTensor * add gather for layernorm * polish * polish		2022-04-28 10:55:40 +08:00
..
_ops	[tensor] refine linear and add gather for laynorm (#893)	2022-04-28 10:55:40 +08:00
__init__.py	[tensor] refine linear and add gather for laynorm (#893)	2022-04-28 10:55:40 +08:00
colo_tensor.py	[tensor] refine linear and add gather for laynorm (#893)	2022-04-28 10:55:40 +08:00
op_wrapper.py	[tensor] reorganize files (#820)	2022-04-21 14:15:48 +08:00
spec.py	[tensor] refine linear and add gather for laynorm (#893)	2022-04-28 10:55:40 +08:00
utils.py	[Tensor] get named parameters for model using ColoTensors (#874)	2022-04-26 14:08:01 +08:00