tf.contrib.framework.fuse_op

View source on GitHub

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

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.

The GraphDef of the new graph.

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