tf.contrib.graph_editor.get_forward_walk_ops
Do a forward graph walk and return all the visited ops.
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 )
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 of tf.Operation within which the search is restricted. If within_ops is None , 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 |
a util.ControlOutputs instance or None. If not None , it will be used while walking the graph forward. |
Returns | |
---|---|
A Python set of all the tf.Operation ahead of seed_ops . |
Raises | |
---|---|
TypeError |
if seed_ops or within_ops cannot be converted to a list of tf.Operation . |
© 2020 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/versions/r1.15/api_docs/python/tf/contrib/graph_editor/get_forward_walk_ops