|
|
@ -257,6 +257,7 @@ from .tensor.stat import numel #DEFINE_ALIAS
|
|
|
|
from .device import get_cudnn_version
|
|
|
|
from .device import get_cudnn_version
|
|
|
|
from .device import set_device
|
|
|
|
from .device import set_device
|
|
|
|
from .device import get_device
|
|
|
|
from .device import get_device
|
|
|
|
|
|
|
|
from .device import is_compiled_with_cuda #DEFINE_ALIAS
|
|
|
|
from .device import is_compiled_with_xpu
|
|
|
|
from .device import is_compiled_with_xpu
|
|
|
|
from .device import XPUPlace
|
|
|
|
from .device import XPUPlace
|
|
|
|
# from .tensor.tensor import Tensor #DEFINE_ALIAS
|
|
|
|
# from .tensor.tensor import Tensor #DEFINE_ALIAS
|
|
|
|