tf.contrib.graph_editor.get_walks_union_ops

View source on GitHub

Return the union of a forward and a backward walk.

tf.contrib.graph_editor.get_walks_union_ops(
    forward_seed_ops, backward_seed_ops, forward_inclusive=True,
    backward_inclusive=True, within_ops=None, within_ops_fn=None,
    control_inputs=False, control_outputs=None, control_ios=None
)

forward_seed_ops an iterable of operations from which the forward graph walk starts. If a list of tensors is given instead, the seed_ops are set to be the consumers of those tensors.
backward_seed_ops an iterable of operations from which the backward graph walk starts. If a list of tensors is given instead, the seed_ops are set to be the generators of those tensors.
forward_inclusive if True the given forward_seed_ops are also part of the resulting set.
backward_inclusive if True the given backward_seed_ops are also part of the resulting set.
within_ops restrict the search within those operations. If within_ops is None, the search is done within the whole graph.
within_ops_fn if provided, a function on ops that should return True iff the op is within the graph traversal. This can be used along within_ops, in which case an op is within if it is also in within_ops.
control_inputs A boolean indicating whether control inputs are enabled.
control_outputs An instance of util.ControlOutputs or None. If not None, control outputs are enabled.
control_ios An instance of util.ControlOutputs or None. If not None, both control inputs and control outputs are enabled. This is equivalent to set control_inputs to True and control_outputs to the util.ControlOutputs instance.

A Python set of all the tf.Operation in the union of a forward and a backward walk.

TypeError if forward_seed_ops or backward_seed_ops or within_ops cannot be converted to a list of tf.Operation.