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