Commit 4be217fc authored by Yuanzhong Xu's avatar Yuanzhong Xu Committed by TensorFlower Gardener
Browse files

[XLA] Allow using specific device_assignment in HloTestBase; adding a... 

[XLA] Allow using specific device_assignment in HloTestBase; adding a static_device_assignment field to HloModuleConfig.

PiperOrigin-RevId: 238087217
parent 4b896eaf
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment