tf.contrib.graph_editor.get_forward_walk_ops(
seed_ops,
inclusive=True,
within_ops=None,
within_ops_fn=None,
stop_at_ts=(),
control_outputs=None
)
Defined in tensorflow/contrib/graph_editor/select.py
.
Do a forward graph walk and return all the visited ops.
Args:
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.inclusive
: if True the given seed_ops are also part of the resulting set.within_ops
: an iterable oftf.Operation
within which the search is restricted. Ifwithin_ops
isNone
, the search is performed 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.stop_at_ts
: an iterable of tensors at which the graph walk stops.control_outputs
: autil.ControlOutputs
instance or None. If notNone
, it will be used while walking the graph forward.
Returns:
A Python set of all the tf.Operation
ahead of seed_ops
.
Raises:
TypeError
: ifseed_ops
orwithin_ops
cannot be converted to a list oftf.Operation
.