Searched refs:_group_value_by_device (Results 1 – 1 of 1) sorted by relevance
402 def _group_value_by_device(per_replica_values): function720 grouped = _group_value_by_device(dense_values)888 device_grads = _group_value_by_device(per_replica_values)1024 grouped_by_device = _group_value_by_device(per_replica_values)