View source on GitHub |
TensorFlow Graph Editor.
Modules
edit
module: Various function for graph editing.
reroute
module: Various function for graph rerouting.
select
module: Various ways of selecting operations and tensors in a graph.
subgraph
module: SubGraphView: a subgraph view on an existing tf.Graph.
transform
module: Class to transform an subgraph into another.
util
module: Utility functions for the graph_editor.
Classes
class ControlOutputs
: The control outputs topology.
class SubGraphView
: A subgraph view on an existing tf.Graph
.
class Transformer
: Transform a subgraph into another one.
class TransformerInfo
: "Contains information about the result of a transform operation.
Functions
add_control_inputs(...)
: Add the control inputs cops to op.
assign_renamed_collections_handler(...)
: Add the transformed elem to the (renamed) collections of elem.
bypass(...)
: Bypass the given subgraph by connecting its inputs to its outputs.
can_be_regex(...)
: Return True if obj can be turned into a regular expression.
check_cios(...)
: Do various check on control_inputs and control_outputs.
compute_boundary_ts(...)
: Compute the tensors at the boundary of a set of ops.
connect(...)
: Connect the outputs of sgv0 to the inputs of sgv1.
copy(...)
: Copy a subgraph.
copy_op_handler(...)
: Copy a tf.Operation
.
copy_with_input_replacements(...)
: Copy a subgraph, replacing some of its inputs.
detach(...)
: Detach both the inputs and the outputs of a subgraph view.
detach_control_inputs(...)
: Detach all the external control inputs of the subgraph sgv.
detach_control_outputs(...)
: Detach all the external control outputs of the subgraph sgv.
detach_inputs(...)
: Detach the inputs of a subgraph view.
detach_outputs(...)
: Detach the output of a subgraph view.
filter_ops(...)
: Get the ops passing the given filter.
filter_ops_from_regex(...)
: Get all the operations that match the given regex.
filter_ts(...)
: Get all the tensors which are input or output of an op in ops.
filter_ts_from_regex(...)
: Get all the tensors linked to ops that match the given regex.
get_backward_walk_ops(...)
: Do a backward graph walk and return all the visited ops. (deprecated)
get_consuming_ops(...)
: Return all the consuming ops of the tensors in ts.
get_forward_walk_ops(...)
: Do a forward graph walk and return all the visited ops.
get_generating_ops(...)
: Return all the generating ops of the tensors in ts
.
get_name_scope_ops(...)
: Get all the operations under the given scope path.
get_ops_ios(...)
: Return all the tf.Operation
which are connected to an op in ops.
get_tensors(...)
: get all the tensors which are input or output of an op in the graph.
get_walks_intersection_ops(...)
: Return the intersection of a forward and a backward walk.
get_walks_union_ops(...)
: Return the union of a forward and a backward walk.
get_within_boundary_ops(...)
: Return all the tf.Operation
within the given boundary.
graph_replace(...)
: Create a new graph which compute the targets from the replaced Tensors.
keep_t_if_possible_handler(...)
: Transform a tensor into itself (identity) if possible.
make_list_of_op(...)
: Convert ops to a list of tf.Operation
.
make_list_of_t(...)
: Convert ts to a list of tf.Tensor
.
make_placeholder_from_dtype_and_shape(...)
: Create a tf.compat.v1.placeholder for the Graph Editor.
make_placeholder_from_tensor(...)
: Create a tf.compat.v1.placeholder
for the Graph Editor.
make_regex(...)
: Return a compiled regular expression.
make_view(...)
: Create a SubGraphView from selected operations and passthrough tensors.
make_view_from_scope(...)
: Make a subgraph from a name scope.
ph(...)
: Create a tf.compat.v1.placeholder for the Graph Editor.
placeholder_name(...)
: Create placeholder name for the graph editor.
remove_control_inputs(...)
: Remove the control inputs cops from co.
replace_t_with_placeholder_handler(...)
: Transform a tensor into a placeholder tensor.
reroute_inputs(...)
: Re-route all the inputs of two subgraphs.
reroute_ios(...)
: Re-route the inputs and outputs of sgv0 to sgv1 (see _reroute_sgv).
reroute_outputs(...)
: Re-route all the outputs of two operations.
reroute_ts(...)
: For each tensor's pair, replace the end of t1 by the end of t0.
select_ops(...)
: Helper to select operations.
select_ops_and_ts(...)
: Helper to select operations and tensors.
select_ts(...)
: Helper to select tensors.
sgv(...)
: Create a SubGraphView from selected operations and passthrough tensors.
sgv_scope(...)
: Make a subgraph from a name scope.
swap_inputs(...)
: Swap all the inputs of sgv0 and sgv1 (see reroute_inputs).
swap_ios(...)
: Swap the inputs and outputs of sgv1 to sgv0 (see _reroute_sgv).
swap_outputs(...)
: Swap all the outputs of sgv0 and sgv1 (see reroute_outputs).
swap_ts(...)
: For each tensor's pair, swap the end of (t0,t1).
transform_op_if_inside_handler(...)
: Transform an optional op only if it is inside the subgraph.