Searched refs:per_replica_values (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 402 def _group_value_by_device(per_replica_values): argument
420 destinations = per_replica_values[0].devices
422 for per_replica_value in per_replica_values:
694 def _batch_all_reduce(self, reduce_op, per_replica_values): argument
697 cross_device_utils.split_by_sparsity(per_replica_values))
878 def _batch_all_reduce(self, reduce_op, per_replica_values): argument
885 (len(per_replica_values), self._all_reduce_spec, self._num_packs,
888 device_grads = _group_value_by_device(per_replica_values)
924 return _ungroup_and_make_mirrored(aggregated_grads, per_replica_values[0],
1017 def _batch_all_reduce(self, reduce_op, per_replica_values): argument
[all …]
|