According to this it’s only available in tf-nightly, which can be installed using this:
pip install tf-nightly
Edit:
Just keeping the answer up to date, updating the tensorflow version also will solve the issue.
You can update it using pip install --upgrade tensorflow
CLICK HERE to find out more related problems solutions.