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 opoutput_quantized
: A boolean flag that indicates if output is quantizedop_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.