aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/c/python_api.h
diff options
context:
space:
mode:
authorGravatar Olivia Nordquist <nolivia@google.com>2017-07-18 11:34:24 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-07-18 11:40:59 -0700
commit9867119831afa821aa53206cc5b7ae773db4284d (patch)
treea50ffe4e4e06d8b932721b7d501c39e7b3c6b559 /tensorflow/c/python_api.h
parent1558c10c8d07c78a58d74d0fcf53c1d2e2505d5e (diff)
Implementing set_device for the C API
PiperOrigin-RevId: 162379684
Diffstat (limited to 'tensorflow/c/python_api.h')
-rw-r--r--tensorflow/c/python_api.h2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/c/python_api.h b/tensorflow/c/python_api.h
index bb2b0fd8fd..e1a55d7755 100644
--- a/tensorflow/c/python_api.h
+++ b/tensorflow/c/python_api.h
@@ -25,6 +25,8 @@ namespace tensorflow {
void AddControlInput(TF_Graph* graph, TF_Operation* op, TF_Operation* input);
+void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device);
+
} // namespace tensorflow
#endif // THIRD_PARTY_TENSORFLOW_C_PYTHON_API_H_