TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.contrib.graph_editor.assign_renamed_collections_handler

View source on GitHub

Add the transformed elem to the (renamed) collections of elem.

tf.contrib.graph_editor.assign_renamed_collections_handler(
    info,
    elem,
    elem_
)

A collection is renamed only if is not a known key, as described in tf.compat.v1.GraphKeys.

Args:

  • info: Transform._TmpInfo instance.
  • elem: the original element (tf.Tensor or tf.Operation)
  • elem_: the transformed element