Fix BERT benchmark test.

PiperOrigin-RevId: 273795511

Author: tensorflower-gardener

official/benchmark/bert_benchmark.py

@@ -76,7 +76,7 @@ def _run_bert_classifier(self, callbacks=None, use_ds=True):
 
     steps_per_loop = 1
 
-    run_classifier.run_customized_training(
+    run_classifier.run_bert_classifier(
         strategy,
         bert_config,
         input_meta_data,