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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/framework/fuse_op