tree: a19c51dd456e63ca947d88353b45a629eec749a0 [path history] [tgz]
  1. __init__.py
  2. BUILD
  3. checkpointing_test.py
  4. collective_all_reduce_strategy_test.py
  5. ctl_correctness_test.py
  6. custom_training_loop_metrics_test.py
  7. custom_training_loop_models_test.py
  8. custom_training_loop_optimizer_test.py
  9. dataset_creator_model_fit_test.py
  10. distribute_strategy_test.py
  11. distributed_file_utils.py
  12. distributed_file_utils_test.py
  13. distributed_training_utils.py
  14. distributed_training_utils_test.py
  15. distributed_training_utils_v1.py
  16. keras_correctness_test_base.py
  17. keras_dnn_correctness_test.py
  18. keras_embedding_model_correctness_test.py
  19. keras_image_model_correctness_test.py
  20. keras_metrics_test.py
  21. keras_models_test.py
  22. keras_optimizer_v2_test.py
  23. keras_premade_models_test.py
  24. keras_rnn_model_correctness_test.py
  25. keras_save_load_test.py
  26. keras_stateful_lstm_model_correctness_test.py
  27. keras_utils_test.py
  28. minimize_loss_test.py
  29. mirrored_strategy_test.py
  30. mirrored_variable_test.py
  31. model_collection_base.py
  32. model_combinations.py
  33. multi_worker_callback_tf2_test.py
  34. multi_worker_test.py
  35. multi_worker_testing_utils.py
  36. optimizer_combinations.py
  37. parameter_server_training_test.py
  38. README.md
  39. saved_model_mixed_api_test.py
  40. saved_model_save_load_test.py
  41. saved_model_test_base.py
  42. sidecar_evaluator.py
  43. sidecar_evaluator_test.py
  44. simple_models.py
  45. strategy_combinations.py
  46. test_example.py
  47. tpu_strategy_test_utils.py
  48. worker_training_state.py
  49. worker_training_state_test.py
tensorflow/python/keras/distribute/README.md

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.