tf.contrib.graph_editor.reroute_ts

View source on GitHub

For each tensor's pair, replace the end of t1 by the end of t0.

tf.contrib.graph_editor.reroute_ts(
    ts0,
    ts1,
    can_modify=None,
    cannot_modify=None
)

B0 B1     B0 B1
|  |    => |/
A0 A1     A0 A1

The end of the tensors in ts1 are left dangling.

Args:

• ts0: an object convertible to a list of tf.Tensor.
• ts1: an object convertible to a list of tf.Tensor.
• can_modify: iterable of operations which can be modified. Any operation outside within_ops will be left untouched by this function.
• cannot_modify: iterable of operations which cannot be modified. Any operation within cannot_modify will be left untouched by this function.

Returns:

The number of individual modifications made by the function.

Raises:

• TypeError: if ts0 or ts1 cannot be converted to a list of tf.Tensor.
• TypeError: if can_modify or cannot_modify is not None and cannot be converted to a list of tf.Operation.