Searched refs:ps_strategy (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | device_setter.py | 76 ps_strategy): argument
95 self._ps_strategy = ps_strategy
136 cluster=None, ps_ops=None, ps_strategy=None): argument
217 if ps_strategy is None:
218 ps_strategy = _RoundRobinStrategy(ps_tasks)
219 if not six.callable(ps_strategy):
222 ps_tasks, ps_device, worker_device, merge_devices, ps_ops, ps_strategy)
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | device_setter_test.py | 40 ps_strategy = device_setter_lib.RandomStrategy(2, seed=0)
44 ps_strategy=ps_strategy)):
63 ps_strategy = device_setter_lib.RandomStrategy(2, seed=0)
64 ps_task = ps_strategy(op)
78 ps_strategy=device_setter_lib.GreedyLoadBalancingStrategy(
99 ps_strategy=device_setter_lib.GreedyLoadBalancingStrategy(
120 ps_strategy=device_setter_lib.GreedyLoadBalancingStrategy(
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 1288 ps_strategy = _OpRoundRobinStrategy(ps_ops, ps_tasks)
1294 ps_strategy=ps_strategy)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.pbtxt | 421 …ce\', \'worker_device\', \'merge_devices\', \'cluster\', \'ps_ops\', \'ps_strategy\'], varargs=Non…
|