Searched defs:distribution_strategy (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | distributed_training_utils.py | 46 def set_weights(distribution_strategy, dist_model, weights): argument
74 def unwrap_values(distribution_strategy, grouped_inputs, grouped_outputs, argument
139 def flatten_perdevice_values(distribution_strategy, perdevice_values): argument
216 def validate_distributed_dataset_inputs(distribution_strategy, x, y, argument
266 def validate_per_device_inputs(distribution_strategy, x): argument
354 def validate_inputs(x, y, distribution_strategy, allow_partial_batch=False): argument
389 def global_batch_size_supported(distribution_strategy): argument
406 def get_input_params(distribution_strategy, first_x_value, steps, batch_size, argument
518 def get_iterator(dataset, distribution_strategy): argument
525 def initialize_iterator(iterator, distribution_strategy): argument
|
| D | training_arrays.py | 502 def _get_iterator(inputs, distribution_strategy=None): argument
509 def _reinitialize_iterator(iterator, distribution_strategy=None): argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 5328 def configure_and_create_distributed_session(distribution_strategy): argument
5331 def _create_session(distribution_strategy): argument
|