dmlc--dgl
Minjie Wang
3bcb268a36
* dist index chapter * preproc chapter * rst * tools page * partition chapter * rst * hetero chapter * 7.1 step1 * add parmetis back * changed based on feedback * address comments
17 行
665 B
Python
17 行
665 B
Python
"""DGL distributed module"""
|
|
from .dist_graph import DistGraphServer, DistGraph, node_split, edge_split
|
|
from .dist_tensor import DistTensor
|
|
from .partition import partition_graph, load_partition, load_partition_feats, load_partition_book
|
|
from .graph_partition_book import GraphPartitionBook, PartitionPolicy
|
|
from .nn import *
|
|
from . import optim
|
|
|
|
from .rpc import *
|
|
from .rpc_server import start_server
|
|
from .rpc_client import connect_to_server, shutdown_servers
|
|
from .dist_context import initialize, exit_client
|
|
from .kvstore import KVServer, KVClient
|
|
from .server_state import ServerState
|
|
from .dist_dataloader import DistDataLoader
|
|
from .graph_services import *
|