Searched refs:variable_device (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy_test.py | 273 variable_device='CPU', argument
289 if 'CPU' in variable_device:
591 strategy, compute_device='CPU', variable_device='CPU', num_gpus=0)
597 strategy, compute_device='GPU', variable_device='GPU', num_gpus=1)
603 strategy, compute_device='GPU', variable_device='CPU', num_gpus=2)
|