Imported from https://github.com/deepset-ai/bert-tensorflow