[TF2XLA] Update the test to match removed XLA:CPU device
PiperOrigin-RevId: 326139696 Change-Id: I65a0341859ea0c3711b7fe7d8b6f47a3d47f6c17
This commit is contained in:
parent
66a07f2509
commit
930274bb8a
@ -36,7 +36,6 @@ DEVICES_PER_TASK = 8
|
||||
|
||||
EXPECTED_DEVICES_PRE_CONNECT = [
|
||||
'/device:CPU:0',
|
||||
'/device:XLA_CPU:0',
|
||||
]
|
||||
EXPECTED_NEW_DEVICES_AFTER_CONNECT_TEMPLATES = [
|
||||
'/job:worker/replica:0/task:{task}/device:CPU:0',
|
||||
|
Loading…
Reference in New Issue
Block a user