Searched defs:distribution_strategy (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distributed_training_utils_v1.py | 51 def set_weights(distribution_strategy, dist_model, weights): argument
79 def unwrap_values(distribution_strategy, grouped_inputs, grouped_outputs, argument
168 def unwrap_outputs(distribution_strategy, grouped_outputs, argument
212 def flatten_per_replica_values(distribution_strategy, per_replica_values): argument
271 def validate_distributed_dataset_inputs(distribution_strategy, x, y, argument
321 def validate_per_replica_inputs(distribution_strategy, x): argument
448 def get_input_params(distribution_strategy, argument
562 def get_iterator(dataset, distribution_strategy): argument
569 def initialize_iterator(iterator, distribution_strategy): argument
|
| D | distributed_training_utils.py | 27 def global_batch_size_supported(distribution_strategy): argument
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_arrays_v1.py | 532 def _get_iterator(inputs, distribution_strategy=None): argument
539 def _reinitialize_iterator(iterator, distribution_strategy=None): argument
|
| D | data_adapter.py | 500 def __init__(self, x, y, steps=None, distribution_strategy=None, **kwargs): argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 6431 def configure_and_create_distributed_session(distribution_strategy): argument
6434 def _create_session(distribution_strategy): argument
|