Returns `model_fn` closure for TPUEstimator.
(bert_config, num_labels, init_checkpoint, learning_rate,
num_train_steps, num_warmup_steps, use_tpu,
use_one_hot_embeddings)
source not stored for this graph (policy: none)