| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | cluster_resolver_test.py | 66 self.assertEqual(resolver.num_accelerators(), {"GPU": 4})
90 self.assertEqual(resolver.num_accelerators(), {"TPU": 4, "GPU": 4})
114 self.assertEqual(resolver.num_accelerators(task_type="worker1", task_id=0),
116 self.assertEqual(resolver.num_accelerators(task_type="worker2", task_id=3),
118 self.assertEqual(resolver.num_accelerators(task_type="worker2", task_id=4),
163 num_accelerators={"GPU": 8},
169 self.assertEqual(simple_resolver.num_accelerators(), {"GPU": 8})
180 num_accelerators={"GPU": 8},
228 num_accelerators={"GPU": 8},
237 num_accelerators={"GPU": 16},
[all …]
|
| D | cluster_resolver.py | 125 def num_accelerators(self, member in ClusterResolver
186 environment='', num_accelerators=None, argument
195 self._num_accelerators = num_accelerators
251 def num_accelerators(self, member in SimpleClusterResolver
443 def num_accelerators(self, member in UnionClusterResolver
447 return self._cluster_resolvers[0].num_accelerators(
|
| D | tfconfig_cluster_resolver.py | 115 def num_accelerators(self, member in TFConfigClusterResolver
121 return super(TFConfigClusterResolver, self).num_accelerators(
|
| D | tfconfig_cluster_resolver_test.py | 284 self.assertEqual(resolver.num_accelerators(), {'TPU': 2, 'GPU': 2})
287 self.assertEqual(resolver.num_accelerators(task_type='worker2', task_id=3),
|
| D | slurm_cluster_resolver.py | 220 def num_accelerators(self, member in SlurmClusterResolver
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.cluster_resolver.-simple-cluster-resolver.pbtxt | 24 …_spec\', \'master\', \'task_type\', \'task_id\', \'environment\', \'num_accelerators\', \'rpc_laye…
35 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-cluster-resolver.pbtxt | 21 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-kubernetes-cluster-resolver.pbtxt | 23 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-slurm-cluster-resolver.pbtxt | 27 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-union-resolver.pbtxt | 35 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-t-f-config-cluster-resolver.pbtxt | 35 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-g-c-e-cluster-resolver.pbtxt | 35 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt | 31 name: "num_accelerators"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.cluster_resolver.-simple-cluster-resolver.pbtxt | 24 …_spec\', \'master\', \'task_type\', \'task_id\', \'environment\', \'num_accelerators\', \'rpc_laye…
35 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-cluster-resolver.pbtxt | 21 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-kubernetes-cluster-resolver.pbtxt | 23 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-slurm-cluster-resolver.pbtxt | 27 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-union-resolver.pbtxt | 35 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-t-f-config-cluster-resolver.pbtxt | 35 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-g-c-e-cluster-resolver.pbtxt | 35 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt | 31 name: "num_accelerators"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | collective_all_reduce_strategy.py | 116 num_gpus = cluster_resolver.num_accelerators().get("GPU", 0)
160 num_gpus = cluster_resolver.num_accelerators().get("GPU", 0)
377 num_accelerators={"GPU": self._num_gpus_per_worker},
|
| D | parameter_server_strategy.py | 135 num_gpus = cluster_resolver.num_accelerators().get("GPU", 0)
216 num_gpus = cluster_resolver.num_accelerators().get("GPU", 0)
487 num_accelerators={"GPU": self._num_gpus_per_worker})
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | collective_all_reduce_strategy.py | 81 num_accelerators={"GPU": num_gpus_per_worker},
|
| D | parameter_server_strategy.py | 161 num_accelerators={'GPU': num_gpus_per_worker})
|