| Name | Last modified | Size | Description | |
|---|---|---|---|---|
| Parent Directory | - | |||
| __init__.py | 2025-01-18 19:26 | 774 | ||
| __pycache__/ | 2025-01-18 19:27 | - | ||
| accelerator_util.py | 2025-01-18 19:26 | 13K | ||
| api.py | 2025-01-18 19:26 | 18K | ||
| config.py | 2025-01-18 19:26 | 8.0K | ||
| d_checkpoint.py | 2025-01-18 19:26 | 21K | ||
| d_variable.py | 2025-01-18 19:26 | 11K | ||
| dtensor_device.py | 2025-01-18 19:26 | 16K | ||
| gen_dtensor_ops.py | 2025-01-18 19:26 | 32K | ||
| input_util.py | 2025-01-18 19:26 | 28K | ||
| layout.py | 2025-01-18 19:26 | 20K | ||
| mesh_util.py | 2025-01-18 19:26 | 12K | ||
| numpy_util.py | 2025-01-18 19:26 | 4.4K | ||
| save_restore.py | 2025-01-18 19:26 | 9.2K | ||
| tests/ | 2025-01-18 19:27 | - | ||
| tpu_util.py | 2025-01-18 19:26 | 33K | ||