Home
last modified time | relevance | path

Searched refs:agg_small_grads_max_group (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops.py573 agg_small_grads_max_group=16): argument
586 if agg_small_grads_max_bytes <= 0 or agg_small_grads_max_group <= 0:
590 self.agg_small_grads_max_group = agg_small_grads_max_group
595 self.agg_small_grads_max_group > 0):
599 max_group=self.agg_small_grads_max_group)
611 agg_small_grads_max_group=0): argument
616 elif agg_small_grads_max_bytes > 0 and agg_small_grads_max_group > 0:
618 agg_small_grads_max_group)
640 agg_small_grads_max_group=10): argument
663 self._agg_small_grads_max_group = agg_small_grads_max_group
[all …]