tf.contrib.graph_editor.copy_with_input_replacements

tf.contrib.graph_editor.copy_with_input_replacements(
    sgv,
    replacement_ts,
    dst_graph=None,
    dst_scope='',
    src_scope='',
    reuse_dst_scope=False
)

Defined in tensorflow/contrib/graph_editor/transform.py.

See the guide: Graph Editor (contrib) > Module: transform

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 TransformerInfo 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.