gigl.distributed.dist_link_prediction_data_partitioner

Classes

DistLinkPredictionDataPartitioner

This class is based on GLT's DistRandomPartitioner class (alibaba/graphlearn-for-pytorch) and has been optimized for better flexibility and memory management.