diff options
| author |  Olivia Nordquist <nolivia@google.com> | 2017-09-26 19:56:26 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-09-26 20:13:58 -0700 |
| commit | c65b9f87d91f51a233cb649f4d1a5b5f63a4d5e1 (patch) | |
| tree | c3c40e0fc0a11857151c1f00c1dd648684d28e50 /tensorflow/c/python_api.cc | |
| parent | 035a9be3cce366ceb57e3bb8d7a436135501061b (diff) | |
implementing _update_input for the C API
PiperOrigin-RevId: 170147211
Diffstat (limited to 'tensorflow/c/python_api.cc')
| -rw-r--r-- | tensorflow/c/python_api.cc | 7 |
1 files changed, 7 insertions, 0 deletions
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 |