Searched refs:dev_prefixes (Results 1 – 1 of 1) sorted by relevance
420 def sum_gradients_all_reduce(dev_prefixes, replica_grads, num_workers, alg, argument438 aux_devices = [prefix + '/cpu:0' for prefix in dev_prefixes]443 for prefix in dev_prefixes