tf.contrib.framework.fuse_op

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

Defined in tensorflow/contrib/framework/python/framework/graph_util.py.

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

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.