Searched defs:num_gpus_per_worker (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | collective_all_reduce_strategy.py | 46 num_gpus_per_worker=0, argument
71 num_gpus_per_worker, argument
|
| D | parameter_server_strategy.py | 78 def __init__(self, num_gpus_per_worker=0): argument
152 def __init__(self, container_strategy, num_gpus_per_worker): argument
|
| D | mirrored_strategy.py | 90 num_gpus_per_worker=None, argument
167 num_gpus_per_worker=None, argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 811 num_gpus_per_worker, argument
953 num_gpus_per_worker=0, argument
|
| D | mirrored_strategy.py | 316 def _cluster_spec_to_device_list(cluster_spec, num_gpus_per_worker): argument
|