Fix batch_norm_benchmark.

PiperOrigin-RevId: 187546384

diff --git a/tensorflow/python/ops/batch_norm_benchmark.py b/tensorflow/python/ops/batch_norm_benchmark.py
index 4f65e3771c08ba55823c9eb71ef5bb3eca0c4aaf..5d68b47aeaef3a90973387ecd5b265eef1e96a5f 100644 (file)
--- a/tensorflow/python/ops/batch_norm_benchmark.py
+++ b/tensorflow/python/ops/batch_norm_benchmark.py
@@ -41,7 +41,7 @@ def batch_norm_op(tensor, mean, variance, beta, gamma, scale):
   # _batch_norm_with_global_normalization is deprecated in v9
   ops.get_default_graph().graph_def_versions.producer = 8
   # pylint: disable=protected-access
-  return gen_nn_ops.batch_norm_with_global_normalization(
+  return gen_nn_ops._batch_norm_with_global_normalization(
       tensor, mean, variance, beta, gamma, 0.001, scale)
   # pylint: enable=protected-access