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