Commit 8e54ffc8 authored by Neal Wu's avatar Neal Wu
Browse files

Add tf.identity

parent 443c0745
......@@ -601,7 +601,7 @@ class BatchNormTest(tf.test.TestCase):
with tf.control_dependencies(update_ops):
barrier = tf.no_op(name='gradient_barrier')
with tf.control_dependencies([barrier]):
output = output
output = tf.identity(output)
# Initialize all variables
sess.run(tf.global_variables_initializer())
moving_mean = variables.get_variables('BatchNorm/moving_mean')[0]
......@@ -632,7 +632,7 @@ class BatchNormTest(tf.test.TestCase):
with tf.control_dependencies(update_ops):
barrier = tf.no_op(name='gradient_barrier')
with tf.control_dependencies([barrier]):
output = output
output = tf.identity(output)
# Initialize all variables
sess.run(tf.global_variables_initializer())
moving_mean = variables.get_variables('BatchNorm/moving_mean')[0]
......@@ -667,7 +667,7 @@ class BatchNormTest(tf.test.TestCase):
with tf.control_dependencies(update_ops):
barrier = tf.no_op(name='gradient_barrier')
with tf.control_dependencies([barrier]):
output = output
output = tf.identity(output)
# Initialize all variables
sess.run(tf.global_variables_initializer())
moving_mean = variables.get_variables('BatchNorm/moving_mean')[0]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment