Can I do graph partition and load partition works with TensorFlow or MXNet backend?

Hi, @VoVAllen , I saw your answer related to using dgl.distributed module with TensorFlow backend.

But I’m still a bit unclear about whether I can use it: Can I do graph partition and load partition tasks using dgl.distributed.partition_graph and dgl.distributed.load_partition with TensorFlow or MXNet backend?

Hope you could help with me! Thanks!