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

Method map_to_global

python/dgl/partition.py:587–593  ·  view source on GitHub ↗

Convert the set of local indices ot global indices

(self, idxs, part_id)

Source from the content-addressed store, hash-verified

585 )
586
587 def map_to_global(self, idxs, part_id):
588 """Convert the set of local indices ot global indices"""
589 return F.zerocopy_from_dgl_ndarray(
590 _CAPI_DGLNDArrayPartitionMapToGlobal(
591 self._partition, F.zerocopy_to_dgl_ndarray(idxs), part_id
592 )
593 )
594
595 def generate_permutation(self, idxs):
596 """Produce a scheme that maps the given indices to separate partitions

Callers 3

get_local_indicesMethod · 0.95
_all_get_tensorMethod · 0.80

Calls

no outgoing calls

Tested by 1