Searched refs:aux_devices (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 383 aux_devices=None, argument
403 scaled_grads, aux_devices, math_ops.add, math_ops.add_n)
405 second_gather_devices = aux_devices[:num_shards]
407 scaled_grads, aux_devices, second_gather_devices, math_ops.add_n)
410 scaled_grads, aux_devices, math_ops.add_n)
438 aux_devices = [prefix + '/cpu:0' for prefix in dev_prefixes]
440 aux_devices = [
446 aux_devices = ['/job:localhost/cpu:0']
449 aux_devices, num_shards if alg_contains_shuffle else 1)
455 grad_and_vars, num_workers, alg, gpu_indices, aux_devices
|