Fix corner case of RANDOM_UNIFORM sampling when no edges exist, and beautify sampling stage name.
Fix corner case of RANDOM_UNIFORM sampling when no edges exist, and beautify sampling stage name.

PiperOrigin-RevId: 444398457

tensorflow/gnn,tensorflow/gnn