HELSON 7a8702c06d [colotensor] add Tensor.view op and its unit test (#1343) ... [colotensor] add megatron initialization for gpt2		2022-07-21 10:53:15 +08:00
..
__init__.py	[checkpoint] checkpoint for ColoTensor Model (#1196)	2022-07-06 17:22:03 +08:00
module_checkpoint.py	[checkpoint] use gather_tensor in checkpoint and update its unit test (#1339)	2022-07-19 14:15:28 +08:00
utils.py	[colotensor] add Tensor.view op and its unit test (#1343)	2022-07-21 10:53:15 +08:00