mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-06 20:04:24 +08:00
975eb8fc5d
* Draft for sparse emb * add some notes * Fix * Add sparse optim for dist pytorch * Update test * Fix * upd * upd * Fix * Fix * Fix bug * add transductive exmpale * Fix example * Some fix * Upd * Fix lint * lint * lint * lint * upd * Fix lint * lint * upd * remove dead import * update * lint * update unitest * update example * Add adam optimizer * Add unitest and update data * upd * upd * upd * Fix docstring and fix some bug in example code * Update rgcn readme Co-authored-by: Ubuntu <ubuntu@ip-172-31-57-25.ec2.internal> Co-authored-by: Ubuntu <ubuntu@ip-172-31-24-210.ec2.internal> Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-66.ec2.internal>
102 lines
2.2 KiB
ReStructuredText
102 lines
2.2 KiB
ReStructuredText
.. _api-distributed:
|
|
|
|
dgl.distributed
|
|
=================================
|
|
|
|
.. automodule:: dgl.distributed
|
|
|
|
Initialization
|
|
---------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
|
|
initialize
|
|
|
|
Distributed Graph
|
|
-----------------
|
|
|
|
.. autoclass:: DistGraph
|
|
:members: ndata, edata, idtype, device, ntypes, etypes, number_of_nodes, number_of_edges, node_attr_schemes, edge_attr_schemes, rank, find_edges, get_partition_book, barrier, local_partition, num_nodes, num_edges, get_node_partition_policy, get_edge_partition_policy, get_etype_id, get_ntype_id, nodes, edges
|
|
|
|
Distributed Tensor
|
|
------------------
|
|
|
|
.. autoclass:: DistTensor
|
|
:members: part_policy, shape, dtype, name
|
|
|
|
Distributed Node Embedding
|
|
---------------------
|
|
.. currentmodule:: dgl.distributed.nn.pytorch
|
|
|
|
.. autoclass:: NodeEmbedding
|
|
|
|
|
|
Distributed embedding optimizer
|
|
-------------------------
|
|
.. currentmodule:: dgl.distributed.optim.pytorch
|
|
|
|
.. autoclass:: SparseAdagrad
|
|
:members: step
|
|
|
|
.. autoclass:: SparseAdam
|
|
:members: step
|
|
|
|
Distributed workload split
|
|
--------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
|
|
node_split
|
|
edge_split
|
|
|
|
Distributed Sampling
|
|
--------------------
|
|
|
|
Distributed DataLoader
|
|
``````````````````````
|
|
|
|
.. currentmodule:: dgl.distributed.dist_dataloader
|
|
|
|
.. autoclass:: DistDataLoader
|
|
|
|
Distributed Neighbor Sampling
|
|
`````````````````````````````
|
|
|
|
.. currentmodule:: dgl.distributed.graph_services
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
|
|
sample_neighbors
|
|
find_edges
|
|
in_subgraph
|
|
|
|
Partition
|
|
---------
|
|
|
|
Graph partition book
|
|
````````````````````
|
|
|
|
.. currentmodule:: dgl.distributed.graph_partition_book
|
|
|
|
.. autoclass:: GraphPartitionBook
|
|
:members: shared_memory, num_partitions, metadata, nid2partid, eid2partid, partid2nids, partid2eids, nid2localnid, eid2localeid, partid, map_to_per_ntype, map_to_per_etype, map_to_homo_nid, map_to_homo_eid
|
|
|
|
.. autoclass:: PartitionPolicy
|
|
:members: policy_str, part_id, partition_book, to_local, to_partid, get_part_size, get_size
|
|
|
|
Split and Load Graphs
|
|
`````````````````````
|
|
|
|
.. currentmodule:: dgl.distributed.partition
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
|
|
load_partition
|
|
load_partition_book
|
|
partition_graph
|
|
|