Ziyue Jiang 9bc5a77c31 [tensor] wrap function in the torch_tensor to ColoTensor (#881)		2022-04-26 20:13:56 +08:00
..
_ops	[Tensor] Add some attributes to ColoTensor (#877)	2022-04-26 15:10:47 +08:00
__init__.py	[Tensor] get named parameters for model using ColoTensors (#874)	2022-04-26 14:08:01 +08:00
colo_tensor.py	[tensor] wrap function in the torch_tensor to ColoTensor (#881)	2022-04-26 20:13:56 +08:00
op_wrapper.py	[tensor] reorganize files (#820)	2022-04-21 14:15:48 +08:00
spec.py	[tensor] customized op returns ColoTensor (#875)	2022-04-26 13:23:59 +08:00
utils.py	[Tensor] get named parameters for model using ColoTensors (#874)	2022-04-26 14:08:01 +08:00