tf.contrib.graph_editor.get_walks_union_ops
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
)
Args |
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. |
Returns |
A Python set of all the tf.Operation in the union of a forward and a backward walk. |
Raises |
TypeError | if forward_seed_ops or backward_seed_ops or within_ops cannot be converted to a list of tf.Operation. |