message: Remove colocate_with in keras layer regularization.
diff: @@ -1311,7 +1311,6 @@ class Layer(checkpointable.CheckpointableBase):
  def _loss_for_variable(v):
  """Creates a regularization loss `Tensor` for variable `v`."""
  - with ops.colocate_with(v):
  with ops.name_scope(name + '/Regularizer'):
  regularization = regularizer(v)
  return regularization
  