TensorFlow 2.0 Beta is available Learn more

tf.contrib.graph_editor.transform_op_if_inside_handler

View source on GitHub

Transform an optional op only if it is inside the subgraph.

tf.contrib.graph_editor.transform_op_if_inside_handler(
    info,
    op,
    keep_if_possible=True
)

This handler is typically use to handle original op: it is fine to keep them if they are inside the subgraph, otherwise they are just ignored.

Args:

  • info: Transform._TmpInfo instance.
  • op: the optional op to transform (or ignore).
  • keep_if_possible: re-attach to the original op if possible, that is, if the source graph and the destination graph are the same.

Returns:

The transformed op or None.