aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/tensorflow.i
diff options
context:
space:
mode:
authorGravatar Eugene Brevdo <ebrevdo@gmail.com>2016-03-09 08:47:40 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-03-09 15:46:12 -0800
commita04bfaed3aa8522f023f91700bb8a33f587ea6ee (patch)
tree17fe44491112406eafd851d6007b20d30d5ec9e8 /tensorflow/python/tensorflow.i
parent53ab80e6eefbb06e9bc9d231d5c7e56cee0558d5 (diff)
Add device_lib swig wrapper which allows accessing the list of local devices.
Change: 116767947
Diffstat (limited to 'tensorflow/python/tensorflow.i')
-rw-r--r--tensorflow/python/tensorflow.i1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/tensorflow.i b/tensorflow/python/tensorflow.i
index 766bdf7dd3..fb2dd00b69 100644
--- a/tensorflow/python/tensorflow.i
+++ b/tensorflow/python/tensorflow.i
@@ -29,5 +29,6 @@ limitations under the License.
%include "tensorflow/python/client/tf_session.i"
%include "tensorflow/python/client/server_lib.i"
+%include "tensorflow/python/client/device_lib.i"
%include "tensorflow/python/framework/python_op_gen.i"