tree: 2be8fb2f238f4e47590d747ef822304dde87a491 [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_ps_only_test.py
  10. dataset_creator_model_fit_test.py
  11. dataset_creator_model_fit_test_base.py
  12. distribute_coordinator_utils.py
  13. distribute_strategy_test.py
  14. distributed_file_utils.py
  15. distributed_file_utils_test.py
  16. distributed_training_utils.py
  17. distributed_training_utils_test.py
  18. distributed_training_utils_v1.py
  19. keras_correctness_test_base.py
  20. keras_dnn_correctness_test.py
  21. keras_embedding_model_correctness_test.py
  22. keras_image_model_correctness_test.py
  23. keras_metrics_test.py
  24. keras_models_test.py
  25. keras_optimizer_v2_test.py
  26. keras_premade_models_test.py
  27. keras_rnn_model_correctness_test.py
  28. keras_save_load_test.py
  29. keras_stateful_lstm_model_correctness_test.py
  30. keras_utils_test.py
  31. minimize_loss_test.py
  32. mirrored_strategy_test.py
  33. mirrored_variable_test.py
  34. model_collection_base.py
  35. model_combinations.py
  36. multi_worker_callback_tf2_test.py
  37. multi_worker_test.py
  38. multi_worker_testing_utils.py
  39. optimizer_combinations.py
  40. parameter_server_evaluation_test.py
  41. README.md
  42. saved_model_mixed_api_test.py
  43. saved_model_save_load_test.py
  44. saved_model_test_base.py
  45. sharded_variable_test.py
  46. sidecar_evaluator.py
  47. sidecar_evaluator_test.py
  48. simple_models.py
  49. strategy_combinations.py
  50. test_example.py
  51. tpu_strategy_test_utils.py
  52. worker_training_state.py
  53. 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.