dmlc--dgl
Hongzhi (Steve), Chen
98325b1097
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
11 行
327 B
Python
11 行
327 B
Python
"""Feature storage classes for DataLoading"""
|
|
from .. import backend as F
|
|
from .base import *
|
|
from .numpy import *
|
|
|
|
# Defines the name TensorStorage
|
|
if F.get_preferred_backend() == "pytorch":
|
|
from .pytorch_tensor import PyTorchTensorStorage as TensorStorage
|
|
else:
|
|
from .tensor import BaseTensorStorage as TensorStorage
|