message: Fix auto_mixed_precision ingraph_train_loop test
diff: @@ -643,10 +643,11 @@ class AutoMixedPrecisionTest(test.TestCase):
  def body(_, i):
  i += 1
  x, yt = it.get_next()
  - y = layers.Dense(nclass)(x)
  + dense = layers.Dense(nclass)
  + y = dense(x)
  loss = losses.sparse_softmax_cross_entropy(yt, y)
  opt = adam.AdamOptimizer()
  - train_op = opt.minimize(loss)
  + train_op = opt.minimize(loss, var_list=dense.trainable_weights)
  with ops.control_dependencies([train_op]):
  loss = array_ops.identity(loss)
  return loss, i
  