TensorFlow 2.0 Beta is available Learn more

tf.contrib.graph_editor.assign_renamed_collections_handler

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

tf.contrib.graph_editor.assign_renamed_collections_handler(
    info,
    elem,
    elem_
)
View source on GitHub

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