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

Update test device names.

Browse Source 
PiperOrigin-RevId: 281631039
Change-Id: I702ead27222687c484c1e479eecf5ab20c448ab4
This commit is contained in:
Revan Sopher 2019-11-20 16:32:33 -08:00 committed by TensorFlower Gardener
parent d79ffabd36
commit 7756864d92
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/python/eager/remote_cloud_tpu_test.py
View File

@ -32,12 +32,12 @@ flags.DEFINE_string('project', None, 'Name of GCP project with TPU.')
flags.DEFINE_string('zone', None, 'Name of GCP zone with TPU.')
EXPECTED_DEVICES_PRE_CONNECT = [
'/job:localhost/replica:0/task:0/device:CPU:0',
'/job:localhost/replica:0/task:0/device:XLA_CPU:0'
'/device:CPU:0',
'/device:XLA_CPU:0',
]
EXPECTED_DEVICES_AFTER_CONNECT = [
'/job:localhost/replica:0/task:0/device:CPU:0',
'/job:localhost/replica:0/task:0/device:XLA_CPU:0',
'/device:CPU:0',
'/device:XLA_CPU:0',
'/job:worker/replica:0/task:0/device:CPU:0',
'/job:worker/replica:0/task:0/device:XLA_CPU:0',
'/job:worker/replica:0/task:0/device:TPU_SYSTEM:0',