TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.contrib.graph_editor.copy_op_handler

View source on GitHub

Copy a tf.Operation.

tf.contrib.graph_editor.copy_op_handler(
    info,
    op,
    new_inputs,
    copy_shape=False,
    nodedef_fn=None
)

Args:

  • info: Transform._TmpInfo instance.
  • op: the tf.Operation to be copied.
  • new_inputs: The new inputs for this op.
  • copy_shape: also copy the shape of the tensor
  • nodedef_fn: If provided, a function that will be run on the NodeDef and should return a mutated NodeDef before a new Operation is created. This is useful as certain features cannot be set on the Operation and must be modified in NodeDef.

Returns:

A (op, op_outputs) tuple containing the transformed op and its outputs.