mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-08 04:24:47 +00:00
[tensor] reorganize files (#820)
This commit is contained in:
7
colossalai/tensor/__init__.py
Normal file
7
colossalai/tensor/__init__.py
Normal 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']
|
Reference in New Issue
Block a user