View source on GitHub |
Sets the current thread device policy.
tf.config.experimental.set_device_policy(
device_policy
)
The device policy controls how operations requiring inputs on a specific device (e.g., on GPU:0) handle inputs on a different device (e.g. GPU:1).
When using the default, an appropriate policy will be picked automatically. The default policy may change over time.
This function only sets the device policy for the current thread. Any subsequently started thread will again use the default policy.
Raises | |
---|---|
ValueError
|
If an invalid device_policy is passed.
|