diff options
Diffstat (limited to 'tensorflow/core/graph/graph.h')
| -rw-r--r-- | tensorflow/core/graph/graph.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/core/graph/graph.h b/tensorflow/core/graph/graph.h index c5dde722fa..d0dba6e1f0 100644 --- a/tensorflow/core/graph/graph.h +++ b/tensorflow/core/graph/graph.h @@ -451,6 +451,11 @@ class Graph { // REQUIRES: The edge must exist. void RemoveEdge(const Edge* edge); + // Removes control edge `edge` from the graph. Note that this also updates + // the corresponding NodeDef to reflect the change. + // REQUIRES: The control edge must exist. + void RemoveControlEdge(const Edge* e); + // Updates the input to a node. The existing edge to `dst` is removed and an // edge from `new_src` to `dst` is created. The NodeDef associated with `dst` // is also updated. |