Searched refs:distribution_strategy (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | distributed_training_utils.py | 46 def set_weights(distribution_strategy, dist_model, weights): argument
67 assign_ops.append(distribution_strategy.unwrap(sw.assign(w)))
74 def unwrap_values(distribution_strategy, grouped_inputs, grouped_outputs, argument
104 all_inputs = flatten_perdevice_values(distribution_strategy,
108 loss = distribution_strategy.reduce(reduce_util.ReduceOp.SUM,
110 all_outputs = flatten_perdevice_values(distribution_strategy,
114 all_outputs = flatten_perdevice_values(distribution_strategy,
118 all_updates = flatten_perdevice_values(distribution_strategy,
128 distribution_strategy, grouped_feed_dict)
133 distribution_strategy, grouped_fetches)
[all …]
|
| D | training_arrays.py | 502 def _get_iterator(inputs, distribution_strategy=None): argument
503 if distribution_strategy:
505 inputs, distribution_strategy)
509 def _reinitialize_iterator(iterator, distribution_strategy=None): argument
510 if distribution_strategy:
512 iterator, distribution_strategy)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | slot_creator.py | 124 distribution_strategy = distribution_strategy_context.get_strategy()
125 with distribution_strategy.extended.colocate_vars_with(primary):
161 distribution_strategy = distribution_strategy_context.get_strategy()
162 with distribution_strategy.extended.colocate_vars_with(primary):
|
| D | sync_replicas_optimizer.py | 263 distribution_strategy = distribution_strategy_context.get_strategy()
264 with distribution_strategy.extended.colocate_vars_with(local_anchor):
|
| D | optimizer.py | 800 distribution_strategy = distribute_ctx.get_strategy()
801 with distribution_strategy.extended.colocate_vars_with(colocate_with):
|
| /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
5341 if is_tpu_strategy(distribution_strategy):
5345 distribution_strategy.configure(session_config)
5346 …master = distribution_strategy.extended._tpu_cluster_resolver.master() # pylint: disable=protecte…
5359 distribution_strategy.configure(session_config)
5367 distribution_strategy,
5370 _create_session(distribution_strategy)
|