[tensor] reorganize files (#820)

This commit is contained in:
Jiarui Fang
2022-04-21 14:15:48 +08:00
committed by GitHub
parent ab962b9735
commit 0ce8924ceb
11 changed files with 71 additions and 76 deletions

View File

@@ -0,0 +1,7 @@
from .op_wrapper import (
colo_op_impl,)
from .colo_tensor import ColoTensor
from .utils import convert_parameter
from ._ops import *
__all__ = ['ColoTensor', 'convert_parameter', 'colo_op_impl']