| Name | Last modified | Size | Description | |
|---|---|---|---|---|
| Parent Directory | - | |||
| __init__.py | 2025-02-26 17:10 | 757 | ||
| __pycache__/ | 2025-02-26 17:11 | - | ||
| accelerator_util.py | 2025-02-26 17:10 | 12K | ||
| api.py | 2025-02-26 17:10 | 18K | ||
| config.py | 2025-02-26 17:10 | 7.8K | ||
| d_checkpoint.py | 2025-02-26 17:10 | 20K | ||
| d_variable.py | 2025-02-26 17:10 | 10K | ||
| dtensor_device.py | 2025-02-26 17:10 | 16K | ||
| gen_dtensor_ops.py | 2025-02-26 17:10 | 31K | ||
| input_util.py | 2025-02-26 17:10 | 27K | ||
| layout.py | 2025-02-26 17:10 | 20K | ||
| mesh_util.py | 2025-02-26 17:10 | 12K | ||
| numpy_util.py | 2025-02-26 17:10 | 4.3K | ||
| save_restore.py | 2025-02-26 17:10 | 9.0K | ||
| tests/ | 2025-02-26 17:11 | - | ||
| tpu_util.py | 2025-02-26 17:10 | 32K | ||