message: Rename tf.colocate_with to tf.compat.v1.colocate_with in upgrade script.
  tf.colocate_with is no longer supported as public API in v2. In the upgrade
  script convert it to tf.compat.v1.colocate_with.
diff: @@ -71,6 +71,7 @@ class TFAPIChangeSpec(ast_edits.APIChangeSpec):
  "tf.train.start_queue_runners": "tf.compat.v1.start_queue_runners",
  })
  # pylint: enable=line-too-long
  + self.function_renames["tf.colocate_with"] = "tf.compat.v1.colocate_with"
  # TODO(amitpatankar): Fix the function rename script
  # to handle constants without hardcoding.
  