Searched defs:distribution_strategy (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distributed_training_utils_v1.py | 55 def set_weights(distribution_strategy, dist_model, weights): argument
83 def unwrap_values(distribution_strategy, grouped_inputs, grouped_outputs, argument
172 def unwrap_outputs(distribution_strategy, grouped_outputs, argument
216 def flatten_per_replica_values(distribution_strategy, per_replica_values): argument
275 def validate_distributed_dataset_inputs(distribution_strategy, x, y, argument
325 def validate_per_replica_inputs(distribution_strategy, x): argument
458 def get_input_params(distribution_strategy, argument
572 def get_iterator(dataset, distribution_strategy): argument
579 def initialize_iterator(iterator, distribution_strategy): argument
|
| D | distributed_training_utils.py | 28 def global_batch_size_supported(distribution_strategy): argument
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/keras_examples_benchmarks/ |
| D | mnist_conv_custom_training_benchmark_test.py | 90 batch_size, distribution_strategy): argument
123 distribution_strategy=None, argument
181 distribution_strategy=None): argument
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/ |
| D | distribution_util.py | 86 def get_distribution_strategy(distribution_strategy="mirrored", argument
|
| D | benchmark_util.py | 109 distribution_strategy='off'): argument
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_arrays_v1.py | 536 def _get_iterator(inputs, distribution_strategy=None): argument
543 def _reinitialize_iterator(iterator, distribution_strategy=None): argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 6504 def configure_and_create_distributed_session(distribution_strategy): argument
6507 def _create_session(distribution_strategy): argument
|