Jiarui Fang d01d3b8cb0 colo init context add device attr. (#866)		2022-04-25 14:24:26 +08:00
..
_ops	[Tensor] add layer norm Op (#852)	2022-04-25 11:49:20 +08:00
__init__.py	[tensor] reorganize files (#820)	2022-04-21 14:15:48 +08:00
colo_tensor.py	colo init context add device attr. (#866)	2022-04-25 14:24:26 +08:00
op_wrapper.py	[tensor] reorganize files (#820)	2022-04-21 14:15:48 +08:00
spec.py	[tensor] an initial dea of tensor spec (#865)	2022-04-25 13:33:52 +08:00
utils.py	[tensor] reorganize files (#820)	2022-04-21 14:15:48 +08:00