contrib.graph_editor.get_forward_walk_ops
tf.contrib.graph_editor.get_forward_walk_ops
tf.contrib.graph_editor.get_forward_walk_ops
get_forward_walk_ops( seed_ops, inclusive=True, within_ops=None, stop_at_ts=(), control_outputs=None )
Defined in tensorflow/contrib/graph_editor/select.py
.
See the guide: Graph Editor (contrib) > Module: select
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. -
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
.
© 2017 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/graph_editor/get_forward_walk_ops