Convert the set of global indices to local indices
(self, idxs)
| 577 | return _CAPI_DGLNDArrayPartitionGetPartSize(self._partition, part) |
| 578 | |
| 579 | def map_to_local(self, idxs): |
| 580 | """Convert the set of global indices to local indices""" |
| 581 | return F.zerocopy_from_dgl_ndarray( |
| 582 | _CAPI_DGLNDArrayPartitionMapToLocal( |
| 583 | self._partition, F.zerocopy_to_dgl_ndarray(idxs) |
| 584 | ) |
| 585 | ) |
| 586 | |
| 587 | def map_to_global(self, idxs, part_id): |
| 588 | """Convert the set of local indices ot global indices""" |
no outgoing calls