Shorted version of meethariprasad's https://github.com/tensorflow/hub/pull/444
Shorted version of meethariprasad's https://github.com/tensorflow/hub/pull/444

PiperOrigin-RevId: 285374004

tensorflow/hub,tensorflow/hub