Searched refs:split_grads_by_size (Results 1 – 2 of 2) sorted by relevance
196 def split_grads_by_size(threshold_size, device_grads): function
905 (this_grads, remaining_grads) = cross_device_utils.split_grads_by_size(