tf.contrib.framework.fuse_op

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2020-10-01 UTC.

Args