mesh_tensorflow/bert/bert.py
mesh_tensorflow/transformer/transformer_layers.py

mesh_tensorflow/bert/run_squad.py
mesh_tensorflow/transformer/utils.py

mesh_tensorflow/bert/run_pretraining.py
mesh_tensorflow/transformer/utils.py

mesh_tensorflow/bert/run_classifier.py
mesh_tensorflow/transformer/utils.py