From c65b9f87d91f51a233cb649f4d1a5b5f63a4d5e1 Mon Sep 17 00:00:00 2001
From: Olivia Nordquist <nolivia@google.com>
Date: Tue, 26 Sep 2017 19:56:26 -0700
Subject: implementing _update_input for the C API

PiperOrigin-RevId: 170147211
---
 tensorflow/c/python_api.cc | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'tensorflow/c/python_api.cc')

diff --git a/tensorflow/c/python_api.cc b/tensorflow/c/python_api.cc
index b8d36b8947..0fe85d5d2c 100644
--- a/tensorflow/c/python_api.cc
+++ b/tensorflow/c/python_api.cc
@@ -29,4 +29,11 @@ void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device) {
   op->node.set_requested_device(device);
 }
 
+void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst,
+                TF_Status* status) {
+  mutex_lock l(graph->mu);
+  status->status = graph->graph.UpdateEdge(&new_src.oper->node, new_src.index,
+                                           &dst.oper->node, dst.index);
+}
+
 }  // namespace tensorflow
-- 
cgit v1.2.3