Graph Editor (contrib)
[TOC]
TensorFlow Graph Editor.
The TensorFlow Graph Editor library allows for modification of an existing
tf.Graph
instance in-place.
The author's github username is purpledog.
Library overview
Appending new nodes is the only graph editing operation allowed by the TensorFlow core library. The Graph Editor library is an attempt to allow for other kinds of editing operations, namely, rerouting and transforming.
- rerouting is a local operation consisting in re-plugging existing tensors (the edges of the graph). Operations (the nodes) are not modified by this operation. For example, rerouting can be used to insert an operation adding noise in place of an existing tensor.
- transforming is a global operation consisting in transforming a graph into another. By default, a transformation is a simple copy but it can be customized to achieved other goals. For instance, a graph can be transformed into another one in which noise is added after all the operations of a specific type.
Important: modifying a graph in-place with the Graph Editor must be done
offline
, that is, without any active sessions.
Of course new operations can be appended online but Graph Editor specific operations like rerouting and transforming can currently only be done offline.
Here is an example of what you cannot do:
- Build a graph.
- Create a session and run the graph.
- Modify the graph with the Graph Editor.
- Re-run the graph with the
same
previously created session.
To edit an already running graph, follow these steps:
- Build a graph.
- Create a session and run the graph.
- Save the graph state and terminate the session
- Modify the graph with the Graph Editor.
- create a new session and restore the graph state
- Re-run the graph with the newly created session.
Note that this procedure is very costly because a new session must be created after any modifications. Among other things, it takes time because the entire graph state must be saved and restored again.
Sub-graph
Most of the functions in the Graph Editor library operate on sub-graph. More precisely, they take as input arguments instances of the SubGraphView class (or anything which can be converted to it). Doing so allows the same function to transparently operate on single operations as well as sub-graph of any size.
A subgraph can be created in several ways:
- using a list of ops:
my_sgv = ge.sgv(ops)
- from a name scope:
my_sgv = ge.sgv_scope("foo/bar", graph=tf.get_default_graph())
- using regular expression:
my_sgv = ge.sgv("foo/.*/.*read$", graph=tf.get_default_graph())
Note that the Graph Editor is meant to manipulate several graphs at the same
time, typically during transform or copy operation. For that reason,
to avoid any confusion, the default graph is never used and the graph on
which to operate must always be given explicitly. This is the reason why
graph=tf.get_default_graph()
is used in the code snippets above.
Modules overview
- util: utility functions.
- select: various selection methods of TensorFlow tensors and operations.
- match: TensorFlow graph matching. Think of this as regular expressions for graphs (but not quite yet).
- reroute: various ways of rerouting tensors to different consuming ops like swap or reroute_a2b.
- subgraph: the SubGraphView class, which enables subgraph manipulations in a
TensorFlow
tf.Graph
. - edit: various editing functions operating on subgraphs like detach, connect or bypass.
- transform: the Transformer class, which enables transforming (or simply copying) a subgraph into another one.
Module: util
tf.contrib.graph_editor.make_list_of_op(ops, check_graph=True, allow_graph=True, ignore_ts=False)
Convert ops to a list of tf.Operation
.
Args:
ops
: can be an iterable oftf.Operation
, atf.Graph
or a single operation.check_graph
: ifTrue
check if all the operations belong to the same graph.allow_graph
: ifFalse
atf.Graph
cannot be converted.ignore_ts
: if True, silently ignoretf.Tensor
.
Returns:
A newly created list of tf.Operation
.
Raises:
TypeError
: if ops cannot be converted to a list oftf.Operation
or, ifcheck_graph
isTrue
, if all the ops do not belong to the same graph.
tf.contrib.graph_editor.get_tensors(graph)
get all the tensors which are input or output of an op in the graph.
Args:
graph
: atf.Graph
.
Returns:
A list of tf.Tensor
.
Raises:
TypeError
: if graph is not atf.Graph
.
tf.contrib.graph_editor.make_list_of_t(ts, check_graph=True, allow_graph=True, ignore_ops=False)
Convert ts to a list of tf.Tensor
.
Args:
ts
: can be an iterable oftf.Tensor
, atf.Graph
or a single tensor.check_graph
: ifTrue
check if all the tensors belong to the same graph.allow_graph
: ifFalse
atf.Graph
cannot be converted.ignore_ops
: ifTrue
, silently ignoretf.Operation
.
Returns:
A newly created list of tf.Tensor
.
Raises:
TypeError
: ifts
cannot be converted to a list oftf.Tensor
or, ifcheck_graph
isTrue
, if all the ops do not belong to the same graph.
tf.contrib.graph_editor.get_generating_ops(ts)
Return all the generating ops of the tensors in ts
.
Args:
ts
: a list oftf.Tensor
Returns:
A list of all the generating tf.Operation
of the tensors in ts
.
Raises:
TypeError
: ifts
cannot be converted to a list oftf.Tensor
.
tf.contrib.graph_editor.get_consuming_ops(ts)
Return all the consuming ops of the tensors in ts.
Args:
ts
: a list oftf.Tensor
Returns:
A list of all the consuming tf.Operation
of the tensors in ts
.
Raises:
TypeError
: if ts cannot be converted to a list oftf.Tensor
.
class tf.contrib.graph_editor.ControlOutputs
The control outputs topology.
tf.contrib.graph_editor.ControlOutputs.__init__(graph)
Create a dictionary of control-output dependencies.
Args:
graph
: atf.Graph
.
Returns:
A dictionary where a key is a tf.Operation
instance and the
corresponding value is a list of all the ops which have the key
as one of their control-input dependencies.
Raises:
TypeError
: graph is not atf.Graph
.
tf.contrib.graph_editor.ControlOutputs.get(op)
return the control outputs of op.
tf.contrib.graph_editor.ControlOutputs.get_all()
tf.contrib.graph_editor.ControlOutputs.graph
tf.contrib.graph_editor.ControlOutputs.update()
Update the control outputs if the graph has changed.
tf.contrib.graph_editor.placeholder_name(t=None, scope=None)
Create placeholder name for the graph editor.
Args:
t
: optional tensor on which the placeholder operation's name will be based onscope
: absolute scope with which to prefix the placeholder's name. None means that the scope of t is preserved. "" means the root scope.
Returns:
A new placeholder name prefixed by "geph". Note that "geph" stands for Graph Editor PlaceHolder. This convention allows to quickly identify the placeholder generated by the Graph Editor.
Raises:
TypeError
: if t is not None or a tf.Tensor.
tf.contrib.graph_editor.make_placeholder_from_tensor(t, scope=None)
Create a tf.placeholder
for the Graph Editor.
Note that the correct graph scope must be set by the calling function.
Args:
t
: atf.Tensor
whose name will be used to create the placeholder (see function placeholder_name).scope
: absolute scope within which to create the placeholder. None means that the scope oft
is preserved.""
means the root scope.
Returns:
A newly created tf.placeholder
.
Raises:
TypeError
: ift
is notNone
or atf.Tensor
.
tf.contrib.graph_editor.make_placeholder_from_dtype_and_shape(dtype, shape=None, scope=None)
Create a tf.placeholder for the Graph Editor.
Note that the correct graph scope must be set by the calling function. The placeholder is named using the function placeholder_name (with no tensor argument).
Args:
dtype
: the tensor type.shape
: the tensor shape (optional).scope
: absolute scope within which to create the placeholder. None means that the scope of t is preserved. "" means the root scope.
Returns:
A newly created tf.placeholder.
Module: select
tf.contrib.graph_editor.filter_ts(ops, positive_filter)
Get all the tensors which are input or output of an op in ops.
Args:
ops
: an object convertible to a list oftf.Operation
.positive_filter
: a function deciding whether to keep a tensor or not. IfTrue
, all the tensors are returned.
Returns:
A list of tf.Tensor
.
Raises:
TypeError
: if ops cannot be converted to a list oftf.Operation
.
tf.contrib.graph_editor.filter_ts_from_regex(ops, regex)
Get all the tensors linked to ops that match the given regex.
Args:
ops
: an object convertible to a list of tf.Operation.regex
: a regular expression matching the tensors' name. For example, "^foo(/.*)?:\d+$" will match all the tensors in the "foo" scope.
Returns:
A list of tf.Tensor.
Raises:
TypeError
: if ops cannot be converted to a list of tf.Operation.
tf.contrib.graph_editor.filter_ops(ops, positive_filter)
Get the ops passing the given filter.
Args:
ops
: an object convertible to a list of tf.Operation.positive_filter
: a function deciding where to keep an operation or not. If True, all the operations are returned.
Returns:
A list of selected tf.Operation.
Raises:
TypeError
: if ops cannot be converted to a list of tf.Operation.
tf.contrib.graph_editor.filter_ops_from_regex(ops, regex)
Get all the operations that match the given regex.
Args:
ops
: an object convertible to a list oftf.Operation
.regex
: a regular expression matching the operation's name. For example,"^foo(/.*)?$"
will match all the operations in the "foo" scope.
Returns:
A list of tf.Operation
.
Raises:
TypeError
: if ops cannot be converted to a list oftf.Operation
.
tf.contrib.graph_editor.get_name_scope_ops(ops, scope)
Get all the operations under the given scope path.
Args:
ops
: an object convertible to a list of tf.Operation.scope
: a scope path.
Returns:
A list of tf.Operation.
Raises:
TypeError
: if ops cannot be converted to a list of tf.Operation.
tf.contrib.graph_editor.check_cios(control_inputs=False, control_outputs=None, control_ios=None)
Do various check on control_inputs and control_outputs.
Args:
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 tuple (control_inputs, control_outputs)
where:
control_inputs
is a boolean indicating whether to use control inputs.
control_outputs
is an instance of util.ControlOutputs or None
Raises:
ValueError
: if control_inputs is an instance of util.ControlOutputs but control_outputs is not NoneTypeError
: if control_outputs is not None and is not a util.ControlOutputs.
tf.contrib.graph_editor.get_ops_ios(ops, control_inputs=False, control_outputs=None, control_ios=None)
Return all the tf.Operation
which are connected to an op in ops.
Args:
ops
: an object convertible to a list oftf.Operation
.control_inputs
: A boolean indicating whether control inputs are enabled.control_outputs
: An instance ofutil.ControlOutputs
orNone
. If notNone
, control outputs are enabled.control_ios
: An instance ofutil.ControlOutputs
orNone
. If notNone
, both control inputs and control outputs are enabled. This is equivalent to setcontrol_inputs
toTrue
andcontrol_outputs
to theutil.ControlOutputs
instance.
Returns:
All the tf.Operation
surrounding the given ops.
Raises:
TypeError
: ifops
cannot be converted to a list oftf.Operation
.
tf.contrib.graph_editor.compute_boundary_ts(ops, ambiguous_ts_are_outputs=True)
Compute the tensors at the boundary of a set of ops.
This function looks at all the tensors connected to the given ops (in/out) and classify them into three categories: 1) input tensors: tensors whose generating operation is not in ops. 2) output tensors: tensors whose consumer operations are not in ops 3) inside tensors: tensors which are neither input nor output tensors.
Args:
ops
: an object convertible to a list of tf.Operation.ambiguous_ts_are_outputs
: a tensor can have consumers both inside and outside ops. Such tensors are treated as outside tensor if ambiguous_ts_are_outputs is True, otherwise they are treated as inside tensor.
Returns:
A tuple (outside_input_ts, outside_output_ts, inside_ts)
where:
outside_input_ts
is a Python list of input tensors;
outside_output_ts
is a python list of output tensors;
inside_ts
is a python list of inside tensors.
Raises:
TypeError
: if ops cannot be converted to a list of tf.Operation.
tf.contrib.graph_editor.get_within_boundary_ops(ops, seed_ops, boundary_ops=(), inclusive=True, control_inputs=False, control_outputs=None, control_ios=None)
Return all the tf.Operation
within the given boundary.
Args:
ops
: an object convertible to a list oftf.Operation
. those ops define the set in which to perform the operation (if atf.Graph
is given, it will be converted to the list of all its operations).seed_ops
: the operations from which to start expanding.boundary_ops
: the ops forming the boundary.inclusive
: ifTrue
, the result will also include the boundary ops.control_inputs
: A boolean indicating whether control inputs are enabled.control_outputs
: An instance ofutil.ControlOutputs
orNone
. If notNone
, control outputs are enabled.control_ios
: An instance ofutil.ControlOutputs
orNone
. If notNone
, both control inputs and control outputs are enabled. This is equivalent to set control_inputs to True and control_outputs to theutil.ControlOutputs
instance.
Returns:
All the tf.Operation
surrounding the given ops.
Raises:
TypeError
: ifops
orseed_ops
cannot be converted to a list oftf.Operation
.ValueError
: if the boundary is intersecting with the seeds.
tf.contrib.graph_editor.get_forward_walk_ops(seed_ops, inclusive=True, within_ops=None, stop_at_ts=(), control_outputs=None)
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
.
tf.contrib.graph_editor.get_backward_walk_ops(seed_ops, inclusive=True, within_ops=None, stop_at_ts=(), control_inputs=False)
Do a backward graph walk and return all the visited ops.
Args:
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.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_inputs
: if True, control inputs will be used while moving backward.
Returns:
A Python set of all the tf.Operation
behind seed_ops
.
Raises:
TypeError
: ifseed_ops
orwithin_ops
cannot be converted to a list oftf.Operation
.
tf.contrib.graph_editor.get_walks_intersection_ops(forward_seed_ops, backward_seed_ops, forward_inclusive=True, backward_inclusive=True, within_ops=None, control_inputs=False, control_outputs=None, control_ios=None)
Return the intersection 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
: an iterable of tf.Operation within which the search is restricted. If within_ops is None, the search is performed within the whole graph.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 intersection of a forward and a backward walk.
Raises:
TypeError
: ifforward_seed_ops
orbackward_seed_ops
orwithin_ops
cannot be converted to a list oftf.Operation
.
tf.contrib.graph_editor.get_walks_union_ops(forward_seed_ops, backward_seed_ops, forward_inclusive=True, backward_inclusive=True, within_ops=None, control_inputs=False, control_outputs=None, control_ios=None)
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.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.
tf.contrib.graph_editor.select_ops(*args, **kwargs)
Helper to select operations.
Args:
*args
: list of 1) regular expressions (compiled or not) or 2) (array of)tf.Operation
.tf.Tensor
instances are silently ignored.**kwargs
: 'graph':tf.Graph
in which to perform the regex query.This is required when using regex. 'positive_filter': an elem if selected only ifpositive_filter(elem)
isTrue
. This is optional. 'restrict_ops_regex': a regular expression is ignored if it doesn't start with the substring "(?#ops)".
Returns:
A list of tf.Operation
.
Raises:
TypeError
: if the optional keyword argument graph is not atf.Graph
or if an argument in args is not an (array of)tf.Operation
or an (array of)tf.Tensor
(silently ignored) or a string or a regular expression.ValueError
: if one of the keyword arguments is unexpected or if a regular expression is used without passing a graph as a keyword argument.
tf.contrib.graph_editor.select_ts(*args, **kwargs)
Helper to select tensors.
Args:
*args
: list of 1) regular expressions (compiled or not) or 2) (array of)tf.Tensor
.tf.Operation
instances are silently ignored.**kwargs
: 'graph':tf.Graph
in which to perform the regex query.This is required when using regex. 'positive_filter': an elem if selected only ifpositive_filter(elem)
isTrue
. This is optional. 'restrict_ts_regex': a regular expression is ignored if it doesn't start with the substring "(?#ts)".
Returns:
A list of tf.Tensor
.
Raises:
TypeError
: if the optional keyword argument graph is not atf.Graph
or if an argument in args is not an (array of)tf.Tensor
or an (array of)tf.Operation
(silently ignored) or a string or a regular expression.ValueError
: if one of the keyword arguments is unexpected or if a regular expression is used without passing a graph as a keyword argument.
tf.contrib.graph_editor.select_ops_and_ts(*args, **kwargs)
Helper to select operations and tensors.
Args:
*args
: list of 1) regular expressions (compiled or not) or 2) (array of)tf.Operation
3) (array of) tf.Tensor. Regular expressions matching tensors must start with the comment"(?#ts)"
, for instance:"(?#ts)^foo/.*"
.**kwargs
: 'graph':tf.Graph
in which to perform the regex query.This is required when using regex. 'positive_filter': an elem if selected only ifpositive_filter(elem)
isTrue
. This is optional.
Returns:
A tuple (ops, ts)
where:
ops
is a list of tf.Operation
, and
ts
is a list of tf.Tensor
Raises:
TypeError
: if the optional keyword argument graph is not atf.Graph
or if an argument in args is not an (array of)tf.Tensor
or an (array of)tf.Operation
or a string or a regular expression.ValueError
: if one of the keyword arguments is unexpected or if a regular expression is used without passing a graph as a keyword argument.
Module: subgraph
class tf.contrib.graph_editor.SubGraphView
A subgraph view on an existing tf.Graph
.
An instance of this class is a subgraph view on an existing tf.Graph
.
"subgraph" means that it can represent part of the whole tf.Graph
.
"view" means that it only provides a passive observation and do not to act
on the tf.Graph
. Note that in this documentation, the term "subgraph" is
often used as substitute to "subgraph view".
A subgraph contains:
- a list of input tensors, accessible via the
inputs
property. - a list of output tensors, accessible via the
outputs
property. - and the operations in between, accessible via the "ops" property.
An subgraph can be seen as a function F(i0, i1, ...) -> o0, o1, ... It is a function which takes as input some input tensors and returns as output some output tensors. The computation that the function performs is encoded in the operations of the subgraph.
The tensors (input or output) can be of two kinds:
- connected: a connected tensor connects to at least one operation contained in the subgraph. One example is a subgraph representing a single operation and its inputs and outputs: all the input and output tensors of the op are "connected".
- passthrough: a passthrough tensor does not connect to any operation contained in the subgraph. One example is a subgraph representing a single tensor: this tensor is passthrough. By default a passthrough tensor is present both in the input and output tensors of the subgraph. It can however be remapped to only appear as an input (or output) only.
The input and output tensors can be remapped. For instance, some input tensor
can be omitted. For instance, a subgraph representing an operation with two
inputs can be remapped to only take one input. Note that this does not change
at all the underlying tf.Graph
(remember, it is a view). It means that
the other input is being ignored, or is being treated as "given".
The analogy with functions can be extended like this: F(x,y) is the original
function. Remapping the inputs from [x, y] to just [x] means that the subgraph
now represent the function F_y(x) (y is "given").
The output tensors can also be remapped. For instance, some output tensor can
be omitted. Other output tensor can be duplicated as well. As mentioned
before, this does not change at all the underlying tf.Graph
.
The analogy with functions can be extended like this: F(...)->x,y is the
original function. Remapping the outputs from [x, y] to just [y,y] means that
the subgraph now represent the function M(F(...)) where M is the function
M(a,b)->b,b.
It is useful to describe three other kind of tensors:
- internal: an internal tensor is a tensor connecting operations contained in the subgraph. One example in the subgraph representing the two operations A and B connected sequentially: -> A -> B ->. The middle arrow is an internal tensor.
- actual input: an input tensor of the subgraph, regardless of whether it is listed in "inputs" or not (masked-out).
- actual output: an output tensor of the subgraph, regardless of whether it is listed in "outputs" or not (masked-out).
- hidden input: an actual input which has been masked-out using an input remapping. In other word, a hidden input is a non-internal tensor not listed as a input tensor and one of whose consumers belongs to the subgraph.
- hidden output: a actual output which has been masked-out using an output remapping. In other word, a hidden output is a non-internal tensor not listed as an output and one of whose generating operations belongs to the subgraph.
Here are some useful guarantees about an instance of a SubGraphView:
- the input (or output) tensors are not internal.
- the input (or output) tensors are either "connected" or "passthrough".
- the passthrough tensors are not connected to any of the operation of the subgraph.
Note that there is no guarantee that an operation in a subgraph contributes at all to its inputs or outputs. For instance, remapping both the inputs and outputs to empty lists will produce a subgraph which still contains all the original operations. However, the remove_unused_ops function can be used to make a new subgraph view whose operations are connected to at least one of the input or output tensors.
An instance of this class is meant to be a lightweight object which is not modified in-place by the user. Rather, the user can create new modified instances of a given subgraph. In that sense, the class SubGraphView is meant to be used like an immutable python object.
A common problem when using views is that they can get out-of-sync with the
data they observe (in this case, a tf.Graph
). This is up to the user to
ensure that this doesn't happen. To keep on the safe side, it is recommended
that the life time of subgraph views are kept very short. One way to achieve
this is to use subgraphs within a "with make_sgv(...) as sgv:" Python context.
To alleviate the out-of-sync problem, some functions are granted the right to
modified subgraph in place. This is typically the case of graph manipulation
functions which, given some subgraphs as arguments, can modify the underlying
tf.Graph
. Since this modification is likely to render the subgraph view
invalid, those functions can modify the argument in place to reflect the
change. For instance, calling the function swap_inputs(svg0, svg1) will modify
svg0 and svg1 in place to reflect the fact that their inputs have now being
swapped.
tf.contrib.graph_editor.SubGraphView.__bool__()
Allows for implicit boolean conversion.
tf.contrib.graph_editor.SubGraphView.__copy__()
Create a copy of this subgraph.
Note that this class is a "view", copying it only create another view and
does not copy the underlying part of the tf.Graph
.
Returns:
A new identical instance of the original subgraph view.
tf.contrib.graph_editor.SubGraphView.__enter__()
Allow Python context to minimize the life time of a subgraph view.
A subgraph view is meant to be a lightweight and transient object. A short lifetime will alleviate the "out-of-sync" issue mentioned earlier. For that reason, a SubGraphView instance can be used within a Python context. For example:
from tensorflow.contrib import graph_editor as ge with ge.make_sgv(...) as sgv: print(sgv)
Returns:
Itself.
tf.contrib.graph_editor.SubGraphView.__exit__(exc_type, exc_value, traceback)
tf.contrib.graph_editor.SubGraphView.__init__(inside_ops=(), passthrough_ts=())
Create a subgraph containing the given ops and the "passthrough" tensors.
Args:
inside_ops
: an object convertible to a list oftf.Operation
. This list defines all the operations in the subgraph.passthrough_ts
: an object convertible to a list oftf.Tensor
. This list define all the "passthrough" tensors. A passthrough tensor is a tensor which goes directly from the input of the subgraph to it output, without any intermediate operations. All the non passthrough tensors are silently ignored.
Raises:
TypeError
: if inside_ops cannot be converted to a list oftf.Operation
or ifpassthrough_ts
cannot be converted to a list oftf.Tensor
.
tf.contrib.graph_editor.SubGraphView.__nonzero__()
Allows for implicit boolean conversion.
tf.contrib.graph_editor.SubGraphView.__str__()
tf.contrib.graph_editor.SubGraphView.connected_inputs
The connected input tensors of this subgraph view.
tf.contrib.graph_editor.SubGraphView.connected_outputs
The connected output tensors of this subgraph view.
tf.contrib.graph_editor.SubGraphView.consumers()
Return a Python set of all the consumers of this subgraph view.
tf.contrib.graph_editor.SubGraphView.copy()
Return a copy of itself.
Note that this class is a "view", copying it only create another view and does not copy the underlying part of the tf.Graph.
Returns:
A new instance identical to the original one.
tf.contrib.graph_editor.SubGraphView.find_op_by_name(op_name)
Return the op named op_name.
Args:
op_name
: the name to search for
Returns:
The op named op_name.
Raises:
ValueError
: if the op_name could not be found.AssertionError
: if the name was found multiple time.
tf.contrib.graph_editor.SubGraphView.graph
The underlying tf.Graph
.
tf.contrib.graph_editor.SubGraphView.input_index(t)
Find the input index corresponding to the given input tensor t.
Args:
t
: the input tensor of this subgraph view.
Returns:
The index in the self.inputs list.
Raises:
Error
: if t in not an input tensor.
tf.contrib.graph_editor.SubGraphView.inputs
The input tensors of this subgraph view.
tf.contrib.graph_editor.SubGraphView.is_passthrough(t)
Check whether a tensor is passthrough.
tf.contrib.graph_editor.SubGraphView.op(op_id)
Get an op by its index.
tf.contrib.graph_editor.SubGraphView.ops
The operations in this subgraph view.
tf.contrib.graph_editor.SubGraphView.output_index(t)
Find the output index corresponding to given output tensor t.
Args:
t
: the output tensor of this subgraph view.
Returns:
The index in the self.outputs list.
Raises:
Error
: if t in not an output tensor.
tf.contrib.graph_editor.SubGraphView.outputs
The output tensors of this subgraph view.
tf.contrib.graph_editor.SubGraphView.passthroughs
The passthrough tensors, going straight from input to output.
tf.contrib.graph_editor.SubGraphView.remap(new_input_indices=None, new_output_indices=None)
Remap the inputs and outputs of the subgraph.
Note that this is only modifying the view: the underlying tf.Graph is not affected.
Args:
new_input_indices
: an iterable of integers representing a mapping between the old inputs and the new ones. This mapping can be under-complete and must be without repetitions.new_output_indices
: an iterable of integers representing a mapping between the old outputs and the new ones. This mapping can be under-complete and can have repetitions.
Returns:
A new modified instance of the original subgraph view with remapped inputs and outputs.
tf.contrib.graph_editor.SubGraphView.remap_default(remove_input_map=True, remove_output_map=True)
Remap the inputs and/or outputs to the default mapping.
Args:
remove_input_map
: if True the input map is reset to the default one.remove_output_map
: if True the output map is reset to the default one.
Returns:
A new modified instance of the original subgraph view with its input and/or output mapping reset to the default one.
tf.contrib.graph_editor.SubGraphView.remap_inputs(new_input_indices)
Remap the inputs of the subgraph.
If the inputs of the original subgraph are [t0, t1, t2], remapping to [2,0] will create a new instance whose inputs is [t2, t0].
Note that this is only modifying the view: the underlying tf.Graph
is not
affected.
Args:
new_input_indices
: an iterable of integers representing a mapping between the old inputs and the new ones. This mapping can be under-complete and must be without repetitions.
Returns:
A new modified instance of the original subgraph view with remapped inputs.
tf.contrib.graph_editor.SubGraphView.remap_outputs(new_output_indices)
Remap the output of the subgraph.
If the output of the original subgraph are [t0, t1, t2], remapping to [1,1,0] will create a new instance whose outputs is [t1, t1, t0].
Note that this is only modifying the view: the underlying tf.Graph is not affected.
Args:
new_output_indices
: an iterable of integers representing a mapping between the old outputs and the new ones. This mapping can be under-complete and can have repetitions.
Returns:
A new modified instance of the original subgraph view with remapped outputs.
tf.contrib.graph_editor.SubGraphView.remap_outputs_make_unique()
Remap the outputs so that all the tensors appears only once.
tf.contrib.graph_editor.SubGraphView.remap_outputs_to_consumers()
Remap the outputs to match the number of consumers.
tf.contrib.graph_editor.SubGraphView.remove_unused_ops(control_inputs=True)
Remove unused ops.
Args:
control_inputs
: if True, control inputs are used to detect used ops.
Returns:
A new subgraph view which only contains used operations.
tf.contrib.graph_editor.make_view(*args, **kwargs)
Create a SubGraphView from selected operations and passthrough tensors.
Args:
*args
: list of 1) regular expressions (compiled or not) or 2) (array of)tf.Operation
3) (array of)tf.Tensor
. Those objects will be converted into a list of operations and a list of candidate for passthrough tensors.**kwargs
: keyword graph is used 1) to check that the ops and ts are from the correct graph 2) for regular expression query
Returns:
A subgraph view.
Raises:
TypeError
: if the optional keyword argument graph is not atf.Graph
or if an argument in args is not an (array of)tf.Tensor
or an (array of)tf.Operation
or a string or a regular expression.ValueError
: if one of the keyword arguments is unexpected.
tf.contrib.graph_editor.make_view_from_scope(scope, graph)
Make a subgraph from a name scope.
Args:
scope
: the name of the scope.graph
: thetf.Graph
.
Returns:
A subgraph view representing the given scope.
Module: reroute
tf.contrib.graph_editor.swap_ts(ts0, ts1, can_modify=None, cannot_modify=None)
For each tensor's pair, swap the end of (t0,t1).
B0 B1 B0 B1 | | => X A0 A1 A0 A1
Args:
ts0
: an object convertible to a list oftf.Tensor
.ts1
: an object convertible to a list oftf.Tensor
.can_modify
: iterable of operations which can be modified. Any operation outside within_ops will be left untouched by this function.cannot_modify
: iterable of operations which cannot be modified. Any operation within cannot_modify will be left untouched by this function.
Returns:
The number of individual modifications made by the function.
Raises:
TypeError
: if ts0 or ts1 cannot be converted to a list of tf.Tensor.TypeError
: if can_modify or cannot_modify is not None and cannot be converted to a list of tf.Operation.
tf.contrib.graph_editor.reroute_a2b_ts(ts0, ts1, can_modify=None, cannot_modify=None)
For each tensor's pair, replace the end of t1 by the end of t0.
B0 B1 B0 B1 | | => |/ A0 A1 A0 A1
The end of the tensors in ts1 are left dangling.
Args:
ts0
: an object convertible to a list oftf.Tensor
.ts1
: an object convertible to a list oftf.Tensor
.can_modify
: iterable of operations which can be modified. Any operation outside within_ops will be left untouched by this function.cannot_modify
: iterable of operations which cannot be modified. Any operation within cannot_modify will be left untouched by this function.
Returns:
The number of individual modifications made by the function.
Raises:
TypeError
: if ts0 or ts1 cannot be converted to a list of tf.Tensor.TypeError
: if can_modify or cannot_modify is not None and cannot be converted to a list of tf.Operation.
tf.contrib.graph_editor.reroute_b2a_ts(ts0, ts1, can_modify=None, cannot_modify=None)
For each tensor's pair, replace the end of t0 by the end of t1.
B0 B1 B0 B1 | | => | A0 A1 A0 A1
The end of the tensors in ts0 are left dangling.
Args:
ts0
: an object convertible to a list oftf.Tensor
.ts1
: an object convertible to a list oftf.Tensor
.can_modify
: iterable of operations which can be modified. Any operation outside within_ops will be left untouched by this function.cannot_modify
: iterable of operations which cannot be modified. Any operation within cannot_modify will be left untouched by this function.
Returns:
The number of individual modifications made by the function.
Raises:
TypeError
: if ts0 or ts1 cannot be converted to a list of tf.Tensor.TypeError
: if can_modify or cannot_modify is not None and cannot be converted to a list of tf.Operation.
tf.contrib.graph_editor.swap_inputs(sgv0, sgv1)
Swap all the inputs of sgv0 and sgv1 (see reroute_inputs).
tf.contrib.graph_editor.reroute_a2b_inputs(sgv0, sgv1)
Re-route all the inputs of sgv0 to sgv1 (see reroute_inputs).
tf.contrib.graph_editor.reroute_b2a_inputs(sgv0, sgv1)
Re-route all the inputs of sgv1 to sgv0 (see reroute_inputs).
tf.contrib.graph_editor.swap_outputs(sgv0, sgv1)
Swap all the outputs of sgv0 and sgv1 (see _reroute_outputs).
tf.contrib.graph_editor.reroute_a2b_outputs(sgv0, sgv1)
Re-route all the outputs of sgv0 to sgv1 (see _reroute_outputs).
tf.contrib.graph_editor.reroute_b2a_outputs(sgv0, sgv1)
Re-route all the outputs of sgv1 to sgv0 (see _reroute_outputs).
tf.contrib.graph_editor.swap(sgv0, sgv1)
Swap the inputs and outputs of sgv1 to sgv0 (see _reroute).
tf.contrib.graph_editor.reroute_a2b(sgv0, sgv1)
Re-route the inputs and outputs of sgv0 to sgv1 (see _reroute).
tf.contrib.graph_editor.reroute_b2a(sgv0, sgv1)
Re-route the inputs and outputs of sgv1 to sgv0 (see _reroute).
tf.contrib.graph_editor.remove_control_inputs(op, cops)
Remove the control inputs cops from co.
Warning: this function is directly manipulating the internals of the
tf.Graph
.
Args:
op
: atf.Operation
from which to remove the control inputs.cops
: an object convertible to a list oftf.Operation
.
Raises:
TypeError
: if op is not atf.Operation
.ValueError
: if any cop in cops is not a control input of op.
tf.contrib.graph_editor.add_control_inputs(op, cops)
Add the control inputs cops to co.
Warning: this function is directly manipulating the internals of the tf.Graph.
Args:
op
: a tf.Operation to which the control inputs are added.cops
: an object convertible to a list oftf.Operation
.
Raises:
TypeError
: if op is not a tf.OperationValueError
: if any cop in cops is already a control input of op.
Module: edit
tf.contrib.graph_editor.detach_control_inputs(sgv)
Detach all the external control inputs of the subgraph sgv.
Args:
sgv
: the subgraph view to be detached. This argument is converted to a subgraph using the same rules as the function subgraph.make_view.
tf.contrib.graph_editor.detach_control_outputs(sgv, control_outputs)
Detach all the external control outputs of the subgraph sgv.
Args:
sgv
: the subgraph view to be detached. This argument is converted to a subgraph using the same rules as the function subgraph.make_view.control_outputs
: a util.ControlOutputs instance.
tf.contrib.graph_editor.detach_inputs(sgv, control_inputs=False)
Detach the inputs of a subgraph view.
Args:
sgv
: the subgraph view to be detached. This argument is converted to a subgraph using the same rules as the function subgraph.make_view. Note that sgv is modified in place.control_inputs
: if True control_inputs are also detached.
Returns:
A tuple (sgv, input_placeholders)
where
sgv
is a new subgraph view of the detached subgraph;
input_placeholders
is a list of the created input placeholders.
Raises:
StandardError
: if sgv cannot be converted to a SubGraphView using the same rules than the function subgraph.make_view.
tf.contrib.graph_editor.detach_outputs(sgv, control_outputs=None)
Detach the output of a subgraph view.
Args:
sgv
: the subgraph view to be detached. This argument is converted to a subgraph using the same rules as the function subgraph.make_view. Note that sgv is modified in place.control_outputs
: a util.ControlOutputs instance or None. If not None the control outputs are also detached.
Returns:
A tuple (sgv, output_placeholders)
where
sgv
is a new subgraph view of the detached subgraph;
output_placeholders
is a list of the created output placeholders.
Raises:
StandardError
: if sgv cannot be converted to a SubGraphView using the same rules than the function subgraph.make_view.
tf.contrib.graph_editor.detach(sgv, control_inputs=False, control_outputs=None, control_ios=None)
Detach both the inputs and the outputs of a subgraph view.
Args:
sgv
: the subgraph view to be detached. This argument is converted to a subgraph using the same rules as the function subgraph.make_view. Note that sgv is modified in place.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 tuple (sgv, detached_inputs, detached_outputs)
where:
sgv
is a new subgraph view of the detached subgraph;
detach_inputs
is a list of the created input placeholders;
detach_outputs
is a list of the created output placeholders.
Raises:
StandardError
: if sgv cannot be converted to a SubGraphView using the same rules than the function subgraph.make_view.
tf.contrib.graph_editor.connect(sgv0, sgv1, disconnect_first=False)
Connect the outputs of sgv0 to the inputs of sgv1.
Args:
sgv0
: the first subgraph to have its outputs swapped. This argument is converted to a subgraph using the same rules as the function subgraph.make_view. Note that sgv0 is modified in place.sgv1
: the second subgraph to have its outputs swapped. This argument is converted to a subgraph using the same rules as the function subgraph.make_view. Note that sgv1 is modified in place.disconnect_first
: if True the current outputs of sgv0 are disconnected.
Returns:
A tuple (sgv0, sgv1)
of the now connected subgraphs.
Raises:
StandardError
: if sgv0 or sgv1 cannot be converted to a SubGraphView using the same rules than the function subgraph.make_view.
tf.contrib.graph_editor.bypass(sgv)
Bypass the given subgraph by connecting its inputs to its outputs.
Args:
sgv
: the subgraph view to be bypassed. This argument is converted to a subgraph using the same rules than the function subgraph.make_view. Note that sgv is modified in place.
Returns:
A tuple (sgv, detached_inputs)
where:
sgv
is a new subgraph view of the bypassed subgraph;
detached_inputs
is a list of the created input placeholders.
Raises:
StandardError
: if sgv cannot be converted to a SubGraphView using the same rules than the function subgraph.make_view.
Module: transform
tf.contrib.graph_editor.replace_t_with_placeholder_handler(info, t)
Transform a tensor into a placeholder tensor.
This handler is typically used to transform a subgraph input tensor into a placeholder.
Args:
info
: Transform._Info instance.t
: tensor whose input must be transformed into a place holder.
Returns:
The tensor generated by the newly created place holder.
tf.contrib.graph_editor.keep_t_if_possible_handler(info, t)
Transform a tensor into itself (identity) if possible.
This handler transform a tensor into itself if the source and destination graph are the same. Otherwise it will create a placeholder. This handler is typically used to transform a hidden input tensors.
Args:
info
: Transform._Info instance.t
: tensor whose input must be transformed into a place holder.
Returns:
The tensor generated by the newly created place holder.
tf.contrib.graph_editor.assign_renamed_collections_handler(info, elem, elem_)
Add the transformed elem to the (renamed) collections of elem.
Args:
info
: Transform._Info instance.elem
: the original element (tf.Tensor
ortf.Operation
)elem_
: the transformed element
tf.contrib.graph_editor.transform_op_if_inside_handler(info, op, keep_if_possible=True)
Transform an optional op only if it is inside the subgraph.
This handler is typically use to handle original op: it is fine to keep them if they are inside the subgraph, otherwise they are just ignored.
Args:
info
: Transform._Info instance.op
: the optional op to transform (or ignore).keep_if_possible
: re-attach to the original op if possible, that is, if the source graph and the destination graph are the same.
Returns:
The transformed op or None.
tf.contrib.graph_editor.copy_op_handler(info, op, copy_shape=True)
Copy a tf.Operation
.
Args:
info
: Transform._Info instance.op
: thetf.Operation
to be copied.copy_shape
: also copy the shape of the tensor
Returns:
A copy of op.
tf.contrib.graph_editor.transform_op_in_place(info, op, detach_outputs=False)
Transform a op in-place - experimental!
Transform an operation in place. It reconnects the inputs if they have been modified. if detach_outputs is True, the outputs of op are also detached.
Args:
info
: Transform._Info instance.op
: the op to transform in place.detach_outputs
: if True, the outputs of op are detached, ready for the user to add more operation.
Returns:
The transformed op.
class tf.contrib.graph_editor.Transformer
Transform a subgraph into another one.
By default, the constructor create a transform which copy a subgraph and replaces inputs with placeholders. This behavior can be modified by changing the handlers.
tf.contrib.graph_editor.Transformer.__call__(sgv, dst_graph, dst_scope, src_scope='', reuse_dst_scope=False)
Execute the transformation.
Args:
sgv
: the source subgraph-view.dst_graph
: the destination graph.dst_scope
: the destination scope.src_scope
: the source scope, which specify the path from which the relative path of the transformed nodes are computed. For instance, if src_scope is a/ and dst_scoped is b/, then the node a/x/y will have a relative path of x/y and will be transformed into b/x/y.reuse_dst_scope
: if True the dst_scope is re-used if it already exists. Otherwise, the scope is given a unique name based on the one given by appending an underscore followed by a digit (default).
Returns:
A tuple (sgv, info)
where:
sgv
is the transformed subgraph view;
info
is an instance of Transformer.ResultInfo containing
information about the transform, including mapping between
original and transformed tensors and operations.
Raises:
ValueError
: if the arguments are invalid.
tf.contrib.graph_editor.Transformer.__init__()
Transformer constructor.
The following members can be modified:
transform_op_handler: handle the transformation of a tf.Operation
.
This handler defaults to a simple copy.
assign_collections_handler: handle the assignment of collections.
This handler defaults to assigning new collections created under the
given name-scope.
transform_external_input_handler: handle the transform of the inputs to
the given subgraph. This handler defaults to creating placeholders
instead of the ops just before the input tensors of the subgraph.
transform_external_hidden_input_handler: handle the transform of the
hidden inputs of the subgraph, that is, the inputs which are not listed
in sgv.inputs. This handler defaults to a transform which keep the same
input if the source and destination graphs are the same, otherwise
use placeholders.
transform_original_op_handler: handle the transform of original_op. This
handler defaults to transforming original_op only if they are in the
subgraph, otherwise they are ignored.
tf.contrib.graph_editor.Transformer.new_name(name)
Compute a destination name from a source name.
Args:
name
: the name to be "transformed".
Returns:
The transformed name.
Raises:
ValueError
: if the source scope is used (that is, not an empty string) and the source name does not belong to the source scope.
tf.contrib.graph_editor.copy(sgv, dst_graph=None, dst_scope='', src_scope='', reuse_dst_scope=False)
Copy a subgraph.
Args:
sgv
: the source subgraph-view. This argument is converted to a subgraph using the same rules than the function subgraph.make_view.dst_graph
: the destination graph.dst_scope
: the destination scope.src_scope
: the source scope.reuse_dst_scope
: if True the dst_scope is re-used if it already exists. Otherwise, the scope is given a unique name based on the one given by appending an underscore followed by a digit (default).
Returns:
A tuple (sgv, info)
where:
sgv
is the transformed subgraph view;
info
is an instance of Transformer.ResultInfo containing
information about the transform, including mapping between
original and transformed tensors and operations.
Raises:
TypeError
: ifdst_graph
is not atf.Graph
.StandardError
: if sgv cannot be converted to a SubGraphView using the same rules than the function subgraph.make_view.
tf.contrib.graph_editor.copy_with_input_replacements(sgv, replacement_ts, dst_graph=None, dst_scope='', src_scope='', reuse_dst_scope=False)
Copy a subgraph, replacing some of its inputs.
Note a replacement only happens if the tensor to be replaced is an input of the given subgraph. The inputs of a subgraph can be queried using sgv.inputs.
Args:
sgv
: the source subgraph-view. This argument is converted to a subgraph using the same rules as the function subgraph.make_view.replacement_ts
: dictionary mapping from original tensors to the replaced one.dst_graph
: the destination graph.dst_scope
: the destination scope.src_scope
: the source scope.reuse_dst_scope
: if True the dst_scope is re-used if it already exists. Otherwise, the scope is given a unique name based on the one given by appending an underscore followed by a digit (default).
Returns:
A tuple (sgv, info)
where:
sgv
is the transformed subgraph view;
info
is an instance of Transformer.ResultInfo containing
information about the transform, including mapping between
original and transformed tensors and operations.
Raises:
TypeError
: if dst_graph is not a tf.Graph.StandardError
: if sgv cannot be converted to a SubGraphView using the same rules as the function subgraph.make_view.
tf.contrib.graph_editor.graph_replace(target_ts, replacement_ts, dst_scope='', src_scope='', reuse_dst_scope=False)
Create a new graph which compute the targets from the replaced Tensors.
Args:
target_ts
: a single tf.Tensor or an iterable of tf.Tensor.replacement_ts
: dictionary mapping from original tensors to replaced tensorsdst_scope
: the destination scope.src_scope
: the source scope.reuse_dst_scope
: if True the dst_scope is re-used if it already exists. Otherwise, the scope is given a unique name based on the one given by appending an underscore followed by a digit (default).
Returns:
A single tf.Tensor or a list of target tf.Tensor, depending on
the type of the input argument target_ts
.
The returned tensors are recomputed using the tensors from replacement_ts.
Raises:
ValueError
: if the targets are not connected to replacement_ts.
Module: match
tf.contrib.graph_editor.op_type(op_types, op=None)
Check if an op is of the given type.
Args:
op_types
: tuple of strings containing the types to check against. For instance: ("Add", "Const")op
: the operation to check (or None).
Returns:
if op is not None, return True if the op is of the correct type. if op is None, return a lambda function which does the type checking.
class tf.contrib.graph_editor.OpMatcher
Graph match class.
tf.contrib.graph_editor.OpMatcher.__call__(op)
Evaluate if the op matches or not.
tf.contrib.graph_editor.OpMatcher.__init__(positive_filter)
Graph match constructor.
tf.contrib.graph_editor.OpMatcher.control_input_ops(*args)
Add input matches.
tf.contrib.graph_editor.OpMatcher.input_ops(*args)
Add input matches.
tf.contrib.graph_editor.OpMatcher.output_ops(*args)
Add output matches.
Useful aliases
tf.contrib.graph_editor.ph(dtype, shape=None, scope=None)
Create a tf.placeholder for the Graph Editor.
Note that the correct graph scope must be set by the calling function. The placeholder is named using the function placeholder_name (with no tensor argument).
Args:
dtype
: the tensor type.shape
: the tensor shape (optional).scope
: absolute scope within which to create the placeholder. None means that the scope of t is preserved. "" means the root scope.
Returns:
A newly created tf.placeholder.
tf.contrib.graph_editor.sgv(*args, **kwargs)
Create a SubGraphView from selected operations and passthrough tensors.
Args:
*args
: list of 1) regular expressions (compiled or not) or 2) (array of)tf.Operation
3) (array of)tf.Tensor
. Those objects will be converted into a list of operations and a list of candidate for passthrough tensors.**kwargs
: keyword graph is used 1) to check that the ops and ts are from the correct graph 2) for regular expression query
Returns:
A subgraph view.
Raises:
TypeError
: if the optional keyword argument graph is not atf.Graph
or if an argument in args is not an (array of)tf.Tensor
or an (array of)tf.Operation
or a string or a regular expression.ValueError
: if one of the keyword arguments is unexpected.
tf.contrib.graph_editor.sgv_scope(scope, graph)
Make a subgraph from a name scope.
Args:
scope
: the name of the scope.graph
: thetf.Graph
.
Returns:
A subgraph view representing the given scope.
tf.contrib.graph_editor.ts(*args, **kwargs)
Helper to select tensors.
Args:
*args
: list of 1) regular expressions (compiled or not) or 2) (array of)tf.Tensor
.tf.Operation
instances are silently ignored.**kwargs
: 'graph':tf.Graph
in which to perform the regex query.This is required when using regex. 'positive_filter': an elem if selected only ifpositive_filter(elem)
isTrue
. This is optional. 'restrict_ts_regex': a regular expression is ignored if it doesn't start with the substring "(?#ts)".
Returns:
A list of tf.Tensor
.
Raises:
TypeError
: if the optional keyword argument graph is not atf.Graph
or if an argument in args is not an (array of)tf.Tensor
or an (array of)tf.Operation
(silently ignored) or a string or a regular expression.ValueError
: if one of the keyword arguments is unexpected or if a regular expression is used without passing a graph as a keyword argument.
tf.contrib.graph_editor.ops(*args, **kwargs)
Helper to select operations.
Args:
*args
: list of 1) regular expressions (compiled or not) or 2) (array of)tf.Operation
.tf.Tensor
instances are silently ignored.**kwargs
: 'graph':tf.Graph
in which to perform the regex query.This is required when using regex. 'positive_filter': an elem if selected only ifpositive_filter(elem)
isTrue
. This is optional. 'restrict_ops_regex': a regular expression is ignored if it doesn't start with the substring "(?#ops)".
Returns:
A list of tf.Operation
.
Raises:
TypeError
: if the optional keyword argument graph is not atf.Graph
or if an argument in args is not an (array of)tf.Operation
or an (array of)tf.Tensor
(silently ignored) or a string or a regular expression.ValueError
: if one of the keyword arguments is unexpected or if a regular expression is used without passing a graph as a keyword argument.
class tf.contrib.graph_editor.matcher
Graph match class.
tf.contrib.graph_editor.matcher.__call__(op)
Evaluate if the op matches or not.
tf.contrib.graph_editor.matcher.__init__(positive_filter)
Graph match constructor.
tf.contrib.graph_editor.matcher.control_input_ops(*args)
Add input matches.
tf.contrib.graph_editor.matcher.input_ops(*args)
Add input matches.
tf.contrib.graph_editor.matcher.output_ops(*args)
Add output matches.