[BERT/TensorFlow1.15] Variable rank not defined？ #554

Describe the bug
A clear and concise description of what the bug is.

To Reproduce
https://github.com/NVIDIA/DeepLearningExamples/blob/master/TensorFlow/LanguageModeling/BERT/run_pretraining.py#L217