tf.contrib.graph_editor.get_walks_union_ops

Return the union of a forward and a backward walk.

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.

© 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_walks_union_ops