graph LR
DistributionStrategy["DistributionStrategy"]
DTensorAPI["DTensorAPI"]
TPUReplication["TPUReplication"]
TPUEmbeddingV2["TPUEmbeddingV2"]
One paragraph explaining the functionality which is represented by this graph. What the main flow is and what is its purpose.
The core component for distributing training across multiple devices.
Related Classes/Methods:
API for distributed tensor operations.
Related Classes/Methods:
Handles replication for TPU devices.
Related Classes/Methods:
Version 2 of TPU embedding.
Related Classes/Methods: