View source on GitHub |
Initialize the TPU devices.
tf.experimental.dtensor.initialize_tpu_system(
enable_coordination_service=False
)
This functions performs additional TPU related initialization after
calling dtensor.initialize_multi_client
to initialize multi-client DTensor.
Refer to dtensor.initialize_multi_client
for relevant environment
variables that controls the initialization of multi-client DTensor.
Raises | |
---|---|
RuntimeError
|
If running inside a tf.function. |
NotFoundError
|
If no TPU devices found in eager mode. |