..
__init__.py
BUILD
Disable a few failing tests on py3.8 version
2020-10-21 19:31:09 -07:00
checkpointing_test.py
Update DS tests in keras to use public TF combination symbols.
2020-09-09 19:53:43 -07:00
collective_all_reduce_strategy_test.py
Move mixed precision files out of experimental/ directory.
2020-10-21 15:19:21 -07:00
ctl_correctness_test.py
Cleanup and consolidate the usage of ds.combinations.main().
2020-09-10 19:55:14 -07:00
custom_training_loop_metrics_test.py
Cleanup and consolidate the usage of ds.combinations.main().
2020-09-10 19:55:14 -07:00
custom_training_loop_models_test.py
Cleanup and consolidate the usage of ds.combinations.main().
2020-09-10 19:55:14 -07:00
custom_training_loop_optimizer_test.py
Remove the usage of ds combination.concat, and replace it with raw list concat.
2020-09-10 09:42:52 -07:00
distribute_strategy_test.py
Move mixed precision files out of experimental/ directory.
2020-10-21 15:19:21 -07:00
distributed_training_utils_test.py
move v1-specific distributed_training_utils to distributed_training_utils_v1.py
2020-09-22 11:37:27 -07:00
distributed_training_utils_v1.py
Move v1-specific training_utils to training_utils_v1.py
2020-09-28 10:59:49 -07:00
distributed_training_utils.py
move v1-specific distributed_training_utils to distributed_training_utils_v1.py
2020-09-22 11:37:27 -07:00
keras_correctness_test_base.py
Move mixed precision files out of experimental/ directory.
2020-10-21 15:19:21 -07:00
keras_dnn_correctness_test.py
tf.distribute: Add MultiWorkerMirroredStrategy to keras_image_correctness_test.
2020-09-18 13:37:31 -07:00
keras_embedding_model_correctness_test.py
Add MWMS combinations to keras_embedding_model_correctness_test.
2020-09-21 12:09:32 -07:00
keras_image_model_correctness_test.py
Use non batch all_reduce in sync batch norm to avoid potential deadlock
2020-10-09 14:46:39 -07:00
keras_metrics_test.py
Update keras/distribute test wrt to tpu_strategy_one_step{_one_core}
2020-09-18 10:35:43 -07:00
keras_models_test.py
Update DS tests in keras to use public TF combination symbols.
2020-09-09 19:53:43 -07:00
keras_optimizer_v2_test.py
Update DS tests in keras to use public TF combination symbols.
2020-09-09 19:53:43 -07:00
keras_premade_models_test.py
tf.distribute: Add MultiWorkerMirroredStrategy to keras_premade_models_test
2020-09-18 12:21:07 -07:00
keras_rnn_model_correctness_test.py
Move mixed precision files out of experimental/ directory.
2020-10-21 15:19:21 -07:00
keras_save_load_test.py
Update DS tests in keras to use public TF combination symbols.
2020-09-09 19:53:43 -07:00
keras_stateful_lstm_model_correctness_test.py
tf.distribute: Add MultiWorkerMirroredStrategy to keras_image_correctness_test.
2020-09-18 13:37:31 -07:00
keras_utils_test.py
model.predict for MultiWorkerMirroredStrategy.
2020-10-07 10:05:58 -07:00
minimize_loss_test.py
Update DS tests in keras to use public TF combination symbols.
2020-09-09 19:53:43 -07:00
mirrored_strategy_test.py
Replace keras usages of private function.defun with tf.function
2020-09-15 11:06:13 -07:00
mirrored_variable_test.py
Update DS tests in keras to use public TF combination symbols.
2020-09-09 19:53:43 -07:00
model_collection_base.py
Move keras model related combinations and tests to keras/distribute.
2020-07-08 21:05:49 -07:00
model_combinations.py
Update DS tests in keras to use public TF combination symbols.
2020-09-09 19:53:43 -07:00
multi_worker_callback_tf2_test.py
MultiProcessRunner: Open source multi_process_runner with a OSS backend.
2020-10-15 14:09:23 -07:00
multi_worker_test.py
Move V1 optimizer code to a separate file optimizer_v1.py from the generic utils in optimizers.py.
2020-09-24 05:34:47 -07:00
multi_worker_testing_utils.py
Update model saving test with MultiWorkerMirroredStrategy.
2020-06-08 02:23:55 -07:00
multi_worker_tutorial_test.py
MultiProcessRunner: symbol replacement: barrier->get_barrier
2020-10-07 10:51:25 -07:00
optimizer_combinations.py
Update DS tests in keras to use public TF combination symbols.
2020-09-09 19:53:43 -07:00
parameter_server_training_test.py
Add a KPL test for PSStrategy with precomputed states.
2020-10-21 12:49:34 -07:00
README.md
Add Readme.md for keras/distribute.
2020-02-07 16:17:20 -08:00
saved_model_mixed_api_test.py
Update DS tests in keras to use public TF combination symbols.
2020-09-09 19:53:43 -07:00
saved_model_save_load_test.py
Update DS tests in keras to use public TF combination symbols.
2020-09-09 19:53:43 -07:00
saved_model_test_base.py
Update DS tests in keras to use public TF combination symbols.
2020-09-09 19:53:43 -07:00
simple_models.py
Move keras model related combinations and tests to keras/distribute.
2020-07-08 21:05:49 -07:00
step_fn_test.py
Update DS tests in keras to use public TF combination symbols.
2020-09-09 19:53:43 -07:00
tpu_strategy_test_utils.py
Add distribution strategy tests.
2020-04-23 11:14:01 -07:00
worker_training_state_test.py
Removing skip condition as windows py35 builds are no longer built.
2020-10-21 15:14:05 -07:00
worker_training_state.py
fix the GitHub Issue #43789 that file_io.delete_recursively_v2 not compatible while calling on files on cloud storage.
2020-10-14 12:39:34 -07:00
