|
__pycache__
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
tests
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
__init__.py
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
accelerator_util.py
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
api.py
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
config.py
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
d_checkpoint.py
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
d_variable.py
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
dtensor_device.py
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
gen_dtensor_ops.py
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
input_util.py
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
layout.py
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
mesh_util.py
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
numpy_util.py
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
save_restore.py
|
1.0
|
2024-05-26 19:49:15 +02:00 |
|
tpu_util.py
|
1.0
|
2024-05-26 19:49:15 +02:00 |
