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