| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | cluster_resolver_test.py | 67 self.assertEqual(resolver.num_accelerators(), {"GPU": 4})
91 self.assertEqual(resolver.num_accelerators(), {"TPU": 4, "GPU": 4})
115 self.assertEqual(resolver.num_accelerators(task_type="worker1", task_id=0),
117 self.assertEqual(resolver.num_accelerators(task_type="worker2", task_id=3),
119 self.assertEqual(resolver.num_accelerators(task_type="worker2", task_id=4),
164 num_accelerators={"GPU": 8},
170 self.assertEqual(simple_resolver.num_accelerators(), {"GPU": 8})
181 num_accelerators={"GPU": 8},
229 num_accelerators={"GPU": 8},
238 num_accelerators={"GPU": 16},
[all …]
|
| D | cluster_resolver.py | 134 def num_accelerators(self, member in ClusterResolver
325 environment='', num_accelerators=None, argument
334 self._num_accelerators = num_accelerators
392 def num_accelerators(self, member in SimpleClusterResolver
615 def num_accelerators(self, member in UnionClusterResolver
619 return self._cluster_resolvers[0].num_accelerators(
|
| D | tfconfig_cluster_resolver.py | 144 def num_accelerators(self, member in TFConfigClusterResolver
150 return super(TFConfigClusterResolver, self).num_accelerators(
|
| D | sagemaker_cluster_resolver.py | 154 def num_accelerators(self, task_type=None, task_id=None, config_proto=None): member in SageMakerClusterResolver
158 self).num_accelerators(task_type, task_id, config_proto)
|
| D | tfconfig_cluster_resolver_test.py | 300 self.assertEqual(resolver.num_accelerators(), {'TPU': 2, 'GPU': 2})
303 self.assertEqual(resolver.num_accelerators(task_type='worker2', task_id=3),
|
| D | slurm_cluster_resolver_test.py | 85 self.assertEqual(slurm_cluster_resolver.num_accelerators(), {'GPU': 1})
|
| /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 | 29 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-union-resolver.pbtxt | 35 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-kubernetes-cluster-resolver.pbtxt | 31 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-g-c-e-cluster-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.-slurm-cluster-resolver.pbtxt | 35 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt | 47 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 | 29 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-union-resolver.pbtxt | 35 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-kubernetes-cluster-resolver.pbtxt | 31 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.-slurm-cluster-resolver.pbtxt | 35 name: "num_accelerators"
|
| D | tensorflow.distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt | 47 name: "num_accelerators"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy.py | 210 num_gpus = cluster_resolver.num_accelerators().get("GPU", 0)
295 num_gpus = cluster_resolver.num_accelerators().get("GPU", 0)
622 num_accelerators={"GPU": self._num_gpus_per_worker})
|
| D | collective_all_reduce_strategy.py | 359 num_gpus = cluster_resolver.num_accelerators().get("GPU", 0)
479 num_gpus = cluster_resolver.num_accelerators().get("GPU", 0)
679 num_accelerators={"GPU": self._num_gpus_per_worker},
|
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/ |
| D | tpu_cluster_resolver_test.py | 665 self.assertEqual(cluster_resolver.num_accelerators(), {'TPU': 2})
707 cluster_resolver.num_accelerators()
|