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