Searched refs:avail_devices (Results 1 – 1 of 1) sorted by relevance
55 def aggregate_gradients_using_hierarchical_copy(avail_devices, replica_grads): argument80 num_devices = len(avail_devices)97 with ops.device(avail_devices[group_0_main_device]):104 with ops.device(avail_devices[group_1_main_device]):109 with ops.device(avail_devices[group_0_main_device]):114 with ops.device(avail_devices[group_0_main_device]):116 with ops.device(avail_devices[group_1_main_device]):121 with ops.device(avail_devices[j]):