Convert the set of local indices ot global indices
(self, idxs, part_id)
| 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 |
no outgoing calls