Fix TF Roberta for mixed precision training
# What does this PR do?

This PR fixes the TF Roberta model for mixed precision training and is now aligned with the other models.

# Fixes

#11282
It looks good to me too. Thanks for the PR!