| Name | Last modified | Size | Description | |
|---|---|---|---|---|
| Parent Directory | - | |||
| __init__.py | 2023-10-05 14:35 | 774 | ||
| __pycache__/ | 2023-10-05 14:35 | - | ||
| accelerator_util.py | 2023-10-05 14:35 | 12K | ||
| api.py | 2023-10-05 14:35 | 18K | ||
| config.py | 2023-10-05 14:35 | 7.8K | ||
| d_checkpoint.py | 2023-10-05 14:35 | 21K | ||
| d_variable.py | 2023-10-05 14:35 | 11K | ||
| dtensor_device.py | 2023-10-05 14:35 | 16K | ||
| gen_dtensor_ops.py | 2023-10-05 14:35 | 32K | ||
| input_util.py | 2023-10-05 14:35 | 28K | ||
| layout.py | 2023-10-05 14:35 | 20K | ||
| mesh_util.py | 2023-10-05 14:35 | 12K | ||
| numpy_util.py | 2023-10-05 14:35 | 4.4K | ||
| save_restore.py | 2023-10-05 14:35 | 9.2K | ||
| tests/ | 2023-10-05 14:35 | - | ||
| tpu_util.py | 2023-10-05 14:35 | 33K | ||