| __pycache__ | |||
| client | |||
| ops | |||
| profiler | |||
| __init__.py | |||
| _tpu_estimator_embedding.py | |||
| api.py | |||
| async_checkpoint.py | |||
| bfloat16.py | |||
| datasets.py | |||
| device_assignment.py | |||
| error_handling.py | |||
| feature_column.py | |||
| feature_column_v2.py | |||
| functional.py | |||
| preempted_hook.py | |||
| session_support.py | |||
| tensor_tracer.py | |||
| tensor_tracer_flags.py | |||
| tensor_tracer_pb2.py | |||
| tensor_tracer_report.py | |||
| topology.py | |||
| tpu.py | |||
| tpu_config.py | |||
| tpu_context.py | |||
| tpu_embedding.py | |||
| tpu_embedding_gradient.py | |||
| tpu_embedding_v2.py | |||
| tpu_embedding_v2_utils.py | |||
| tpu_estimator.py | |||
| tpu_feed.py | |||
| tpu_function.py | |||
| tpu_name_util.py | |||
| tpu_optimizer.py | |||
| tpu_sharding.py | |||
| tpu_strategy_util.py | |||
| tpu_system_metadata.py | |||
| training_loop.py | |||
| util.py |
