experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/python/keras/distribute
History
A. Unique TensorFlower 034633f23b PY2 removal cleanup 
PiperOrigin-RevId: 352106691
Change-Id: I382d53c64f0d29da430b8cb6d2395a2cb281509e
2021-01-15 16:48:57 -08:00
..
__init__.py
SidecarEvaluator: Create a Keras SidecarEvaluator API for a dedicated evaluator task.
2020-10-29 02:13:29 -07:00
BUILD
PY2 removal cleanup
2021-01-15 16:48:57 -08: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
Remove outdated MWMS tests
2020-11-10 12:18:37 -08:00
ctl_correctness_test.py
Remove skip condition in ctl_correctness_test for MWMS + XLA. It is already disabled at the strategy combination level.
2020-11-20 12:03:38 -08:00
custom_training_loop_metrics_test.py
Update metrics to better support tensors that are not fully defined
2020-11-11 09:36:16 -08:00
custom_training_loop_models_test.py
Preserve TPUDistributedVariables passed to TPUStrategy.run().
2021-01-13 19:22:09 -08:00
custom_training_loop_optimizer_test.py
Update keras to use tf.__internal__ for distribute tests.
2020-11-08 20:40:18 -08:00
distribute_strategy_test.py
Move the LossReduction class from tf to Keras.
2021-01-13 13:40:14 -08:00
distributed_file_utils_test.py
Move the distribute_file_utils to Keras.
2020-10-26 15:29:19 -07:00
distributed_file_utils.py
Move the distribute_file_utils to Keras.
2020-10-26 15:29:19 -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
Internal symbol name change.
2021-01-05 14:15:45 -08:00
distributed_training_utils.py
[*.py,tensorflow/cc/framework/cc_op_gen.cc] Rename "Arguments:" to "Args:"
2020-12-22 09:24:04 +11:00
keras_correctness_test_base.py
[*.py,tensorflow/cc/framework/cc_op_gen.cc] Rename "Arguments:" to "Args:"
2020-12-22 09:24:04 +11:00
keras_dnn_correctness_test.py
Update keras to use tf.__internal__ for distribute tests.
2020-11-08 20:40:18 -08: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 keras to use tf.__internal__ for distribute tests.
2020-11-08 20:40:18 -08: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
Update keras to use tf.__internal__ for distribute tests.
2020-11-08 20:40:18 -08: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
Cleanup keras distribution tests.
2020-12-01 11:57:31 -08: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
Keras Fixit: Copy the util is_oss into multi_worker_callback_tf2_test.
2020-10-27 20:18:20 -07:00
multi_worker_test.py
Keras Fixit: Remove usage of parameter_server_strategy.ParameterServerStrategyExtended in Keras.
2020-10-27 19:41:51 -07:00
multi_worker_testing_utils.py
Update model saving test with MultiWorkerMirroredStrategy.
2020-06-08 02:23:55 -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
Move the LossReduction class from tf to Keras.
2021-01-13 13:40:14 -08: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
sharded_variable_test.py
Fork keras related sharded_variable_test to keras/distribute.
2020-11-03 11:18:23 -08:00
sidecar_evaluator_test.py
add support to load checkpoint that saved by ModelCheckpoint callback for sidecar_evaluator.
2020-12-17 12:31:09 -08:00
sidecar_evaluator.py
add support to load checkpoint that saved by ModelCheckpoint callback for sidecar_evaluator.
2020-12-17 12:31:09 -08:00
simple_models.py
Move keras model related combinations and tests to keras/distribute.
2020-07-08 21:05:49 -07:00
strategy_combinations.py
Update keras to use tf.__internal__ for distribute tests.
2020-11-08 20:40:18 -08:00
test_example.py
Cleanup keras distribution tests.
2020-12-01 11:57:31 -08:00
tpu_strategy_test_utils.py
Replace tf.flags with absl flags.
2020-10-27 16:28:28 -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
[*.py,tensorflow/cc/framework/cc_op_gen.cc] Rename "Arguments:" to "Args:"
2020-12-22 09:24:04 +11:00

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.