Searched refs:gpu_indices (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 382 gpu_indices, argument
394 scaled_grads, num_workers, num_shards, gpu_indices, math_ops.add)
421 num_shards, gpu_indices): argument
442 for i in range(len(gpu_indices))
455 grad_and_vars, num_workers, alg, gpu_indices, aux_devices
|