HomeSort by relevance Sort by last modified time
    Searched refs:distribution_strategy (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/python/keras/engine/
distributed_training_utils.py 46 def set_weights(distribution_strategy, dist_model, weights):
54 distribution_strategy: DistributionStrategy used to distribute training
67 assign_ops.append(distribution_strategy.unwrap(sw.assign(w)))
74 def unwrap_values(distribution_strategy, grouped_inputs, grouped_outputs,
85 distribution_strategy: DistributionStrategy used to distribute training and
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,
    [all...]
training_arrays.py 502 def _get_iterator(inputs, distribution_strategy=None):
503 if distribution_strategy:
505 inputs, distribution_strategy)
509 def _reinitialize_iterator(iterator, distribution_strategy=None):
510 if distribution_strategy:
512 iterator, distribution_strategy)
  /external/tensorflow/tensorflow/python/training/
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):
sync_replicas_optimizer.py 263 distribution_strategy = distribution_strategy_context.get_strategy()
264 with distribution_strategy.extended.colocate_vars_with(local_anchor):
optimizer.py 800 distribution_strategy = distribute_ctx.get_strategy()
801 with distribution_strategy.extended.colocate_vars_with(colocate_with):
    [all...]
  /external/tensorflow/tensorflow/python/keras/
backend.py     [all...]

Completed in 300 milliseconds