MCPcopy
hub / github.com/dmlc/dgl / nid2partid

Method nid2partid

python/dgl/distributed/graph_partition_book.py:333–347  ·  view source on GitHub ↗

From global node IDs to partition IDs Parameters ---------- nids : tensor global node IDs ntype : str The node type Returns ------- tensor partition IDs

(self, nids, ntype)

Source from the content-addressed store, hash-verified

331 """
332
333 def nid2partid(self, nids, ntype):
334 """From global node IDs to partition IDs
335
336 Parameters
337 ----------
338 nids : tensor
339 global node IDs
340 ntype : str
341 The node type
342
343 Returns
344 -------
345 tensor
346 partition IDs
347 """
348
349 def eid2partid(self, eids, etype):
350 """From global edge IDs to partition IDs

Callers 8

verify_graph_featsFunction · 0.45
verify_node_partitionidsFunction · 0.45
_distributed_accessFunction · 0.45
_verify_dgl_partitionFunction · 0.45
to_partidMethod · 0.45
verify_graph_featsFunction · 0.45
check_partitionFunction · 0.45

Calls

no outgoing calls

Tested by 2

verify_graph_featsFunction · 0.36
check_partitionFunction · 0.36