Searched refs:agg_small_grads_max_bytes (Results 1 – 1 of 1) sorted by relevance
572 agg_small_grads_max_bytes=1048576, argument586 if agg_small_grads_max_bytes <= 0 or agg_small_grads_max_group <= 0:589 self.agg_small_grads_max_bytes = agg_small_grads_max_bytes594 if (self.agg_small_grads_max_bytes > 0 and598 max_bytes=self.agg_small_grads_max_bytes,610 agg_small_grads_max_bytes=0, argument616 elif agg_small_grads_max_bytes > 0 and agg_small_grads_max_group > 0:617 tensor_packer = _AggregateSmallTensorPacker(agg_small_grads_max_bytes,639 agg_small_grads_max_bytes=0, argument662 self._agg_small_grads_max_bytes = agg_small_grads_max_bytes[all …]