tf.contrib.framework.fuse_op

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