STT-tensorflow/tf/tensorflow/python/keras/distribute
2021-01-21 15:40:33 -08:00
..
__init__.py initial 2021-01-21 09:06:36 -08:00
BUILD sync to master 2021-01-21 15:40:33 -08:00
checkpointing_test.py initial 2021-01-21 09:06:36 -08:00
collective_all_reduce_strategy_test.py initial 2021-01-21 09:06:36 -08:00
ctl_correctness_test.py initial 2021-01-21 09:06:36 -08:00
custom_training_loop_metrics_test.py initial 2021-01-21 09:06:36 -08:00
custom_training_loop_models_test.py initial 2021-01-21 09:06:36 -08:00
custom_training_loop_optimizer_test.py initial 2021-01-21 09:06:36 -08:00
distribute_strategy_test.py initial 2021-01-21 09:06:36 -08:00
distributed_file_utils_test.py initial 2021-01-21 09:06:36 -08:00
distributed_file_utils.py initial 2021-01-21 09:06:36 -08:00
distributed_training_utils_test.py initial 2021-01-21 09:06:36 -08:00
distributed_training_utils_v1.py initial 2021-01-21 09:06:36 -08:00
distributed_training_utils.py initial 2021-01-21 09:06:36 -08:00
keras_correctness_test_base.py initial 2021-01-21 09:06:36 -08:00
keras_dnn_correctness_test.py initial 2021-01-21 09:06:36 -08:00
keras_embedding_model_correctness_test.py initial 2021-01-21 09:06:36 -08:00
keras_image_model_correctness_test.py initial 2021-01-21 09:06:36 -08:00
keras_metrics_test.py initial 2021-01-21 09:06:36 -08:00
keras_models_test.py initial 2021-01-21 09:06:36 -08:00
keras_optimizer_v2_test.py initial 2021-01-21 09:06:36 -08:00
keras_premade_models_test.py initial 2021-01-21 09:06:36 -08:00
keras_rnn_model_correctness_test.py initial 2021-01-21 09:06:36 -08:00
keras_save_load_test.py initial 2021-01-21 09:06:36 -08:00
keras_stateful_lstm_model_correctness_test.py initial 2021-01-21 09:06:36 -08:00
keras_utils_test.py initial 2021-01-21 09:06:36 -08:00
minimize_loss_test.py initial 2021-01-21 09:06:36 -08:00
mirrored_strategy_test.py initial 2021-01-21 09:06:36 -08:00
mirrored_variable_test.py initial 2021-01-21 09:06:36 -08:00
model_collection_base.py initial 2021-01-21 09:06:36 -08:00
model_combinations.py initial 2021-01-21 09:06:36 -08:00
multi_worker_callback_tf2_test.py initial 2021-01-21 09:06:36 -08:00
multi_worker_test.py initial 2021-01-21 09:06:36 -08:00
multi_worker_testing_utils.py initial 2021-01-21 09:06:36 -08:00
optimizer_combinations.py initial 2021-01-21 09:06:36 -08:00
parameter_server_training_test.py initial 2021-01-21 09:06:36 -08:00
README.md initial 2021-01-21 09:06:36 -08:00
saved_model_mixed_api_test.py initial 2021-01-21 09:06:36 -08:00
saved_model_save_load_test.py initial 2021-01-21 09:06:36 -08:00
saved_model_test_base.py initial 2021-01-21 09:06:36 -08:00
sharded_variable_test.py initial 2021-01-21 09:06:36 -08:00
sidecar_evaluator_test.py initial 2021-01-21 09:06:36 -08:00
sidecar_evaluator.py initial 2021-01-21 09:06:36 -08:00
simple_models.py initial 2021-01-21 09:06:36 -08:00
strategy_combinations.py initial 2021-01-21 09:06:36 -08:00
test_example.py initial 2021-01-21 09:06:36 -08:00
tpu_strategy_test_utils.py initial 2021-01-21 09:06:36 -08:00
worker_training_state_test.py initial 2021-01-21 09:06:36 -08:00
worker_training_state.py initial 2021-01-21 09:06:36 -08:00

Keras with Distribution Strategy Tests

This directory contains unit tests that combine Keras library with Distribution Training. Tests that use a custom training loop instead of Keras compile/fit should be placed under python/distribute directory instead.