experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Expose tf.config.experimental_list_devices

Browse Source 
PiperOrigin-RevId: 234895637
This commit is contained in:
Alexandre Passos 2019-02-20 16:44:40 -08:00 committed by TensorFlower Gardener
parent 2127948229
commit 3a83826bd1
3 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/python/eager/context.py
View File

@ -921,6 +921,7 @@ def device(name):
return context().device(name)
@tf_export("config.experimental_list_devices")
def list_devices():
"""List the names of the available devices.

4
tensorflow/tools/api/golden/v1/tensorflow.config.pbtxt
View File

@ -4,6 +4,10 @@ tf_module {
name: "experimental_connect_to_host"
argspec: "args=[\'remote_host\', \'job_name\'], varargs=None, keywords=None, defaults=[\'None\', \'worker\'], "
}
member_method {
name: "experimental_list_devices"
argspec: "args=[], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "experimental_run_functions_eagerly"
argspec: "args=[\'run_eagerly\'], varargs=None, keywords=None, defaults=None"

4
tensorflow/tools/api/golden/v2/tensorflow.config.pbtxt
View File

@ -4,6 +4,10 @@ tf_module {
name: "experimental_connect_to_host"
argspec: "args=[\'remote_host\', \'job_name\'], varargs=None, keywords=None, defaults=[\'None\', \'worker\'], "
}
member_method {
name: "experimental_list_devices"
argspec: "args=[], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "experimental_run_functions_eagerly"
argspec: "args=[\'run_eagerly\'], varargs=None, keywords=None, defaults=None"