message: Fixes breaking mirrored_strategy_multigpu_test.
diff: @@ -43,8 +43,11 @@ function = LazyLoader("function", globals(),
  "tensorflow.python.eager.function")
  WHITELIST_COLLECTIONS = [
  - ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.LOCAL_VARIABLES,
  - ops.GraphKeys.TRAINABLE_VARIABLES
  + ops.GraphKeys.GLOBAL_VARIABLES,
  + ops.GraphKeys.LOCAL_VARIABLES,
  + ops.GraphKeys.TRAINABLE_VARIABLES,
  + variable_scope._VARSTORE_KEY, # pylint: disable=protected-access
  + variable_scope._VARSCOPESTORE_KEY # pylint: disable=protected-access
  ]
  @@ -132,16 +135,10 @@ class FuncGraph(ops.Graph):
  self._collections = graph._collections
  else:
  for collection_name in graph.get_all_collection_keys():
  - if collection_name in WHITELIST_COLLECTIONS:
  - self._collections[collection_name] = graph.get_collection_ref(
  - collection_name)
  - else:
  + if collection_name not in WHITELIST_COLLECTIONS:
  self._collections[collection_name] = graph.get_collection(
  collection_name)
  - for collection_name in [
  - variable_scope._VARSTORE_KEY,
  - variable_scope._VARSCOPESTORE_KEY # pylint: disable=protected-access
  - ]:
  + for collection_name in WHITELIST_COLLECTIONS:
  self._collections[collection_name] = graph.get_collection_ref(
  collection_name)
  