r2.14 cherry-pick: 75d692bb9de "Remove deprecated "distributions" package from exports."
Refer to the original commit: https://github.com/tensorflow/tensorflow/commit/75d692bb9def905f2d84970d50e9edb137fbf1a5