TensorFlow 2.0 Beta is available Learn more

tf.contrib.framework.fuse_op

View source on GitHub

Fuse subgraph between input_nodes and output_nodes into a single custom op.

tf.contrib.framework.fuse_op(
    graph_def,
    input_nodes,
    output_nodes,
    output_dtypes,
    output_quantized,
    op_name,
    op_type
)

Args:

  • graph_def: A graph_pb2.GraphDef proto.
  • input_nodes: input nodes to the subgraph to be fused.
  • output_nodes: output nodes to the subgraph to be fused.
  • output_dtypes: A list of output datatypes for the custom op
  • output_quantized: A boolean flag that indicates if output is quantized
  • op_name: fused op name.
  • op_type: fused op type.

Returns:

The GraphDef of the new graph.

Raises:

  • TypeError: If 'graph_def' is not a graph_pb2.GraphDef proto.