TensorFlow 2.0 Beta is available Learn more

Module: tf.contrib.graph_editor.subgraph

SubGraphView: a subgraph view on an existing tf.Graph.

View source on GitHub