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.
|