diff options
author | Olivia Nordquist <nolivia@google.com> | 2017-07-18 11:34:24 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-18 11:40:59 -0700 |
commit | 9867119831afa821aa53206cc5b7ae773db4284d (patch) | |
tree | a50ffe4e4e06d8b932721b7d501c39e7b3c6b559 /tensorflow/c/python_api.h | |
parent | 1558c10c8d07c78a58d74d0fcf53c1d2e2505d5e (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.h | 2 |
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_ |