Add pretraining loss computation for TF Bert pretraining
# What does this PR do?

This PR adds the loss computation for the pretraining TF BERT model. The loss computation test is also more robust on variable call signature lengths.