refactor for r11

control_flow_ops deprecated

Author: bigsnarfdude

python/libs/batch_norm.py

@@ -3,7 +3,6 @@
 """
 
 import tensorflow as tf
-from tensorflow.python import control_flow_ops
 
 
 def batch_norm(x, phase_train, scope='bn', affine=True):
@@ -54,7 +53,7 @@ def mean_var_with_update():
             """
             with tf.control_dependencies([ema_apply_op]):
                 return tf.identity(batch_mean), tf.identity(batch_var)
-        mean, var = control_flow_ops.cond(phase_train,
+        mean, var = tf.cond(phase_train,
                                           mean_var_with_update,
                                           lambda: (ema_mean, ema_var))