See the guide: Graph Editor (contrib) > Module: select
Do a forward graph walk and return all the visited ops.
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.Operationwithin which the search is restricted. If
None, the search is performed within the whole graph.
stop_at_ts: an iterable of tensors at which the graph walk stops.
util.ControlOutputsinstance or None. If not
None, it will be used while walking the graph forward.
A Python set of all the
tf.Operation ahead of
within_opscannot be converted to a list of
© 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.