| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | cluster_resolver.py | 37 def format_master_url(master, rpc_layer=None): argument
38 if rpc_layer:
39 return '%s://%s' % (rpc_layer, master)
108 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
187 rpc_layer=None): argument
196 self._rpc_layer = rpc_layer
210 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
229 return format_master_url(master, rpc_layer=rpc_layer or self._rpc_layer)
274 def rpc_layer(self): member in SimpleClusterResolver
277 @rpc_layer.setter
[all …]
|
| D | tfconfig_cluster_resolver.py | 35 def format_master_url(master, rpc_layer=None): argument
36 if rpc_layer:
37 return '%s://%s' % (rpc_layer, master)
58 rpc_layer=None, argument
73 self._rpc_layer = rpc_layer
105 def rpc_layer(self): member in TFConfigClusterResolver
111 @rpc_layer.setter
112 def rpc_layer(self, rpc_layer): member in TFConfigClusterResolver
113 self._rpc_layer = rpc_layer
135 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
[all …]
|
| D | gce_cluster_resolver.py | 52 rpc_layer='grpc', argument
88 self._rpc_layer = rpc_layer
151 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
157 if rpc_layer or self._rpc_layer:
158 return '%s://%s' % (rpc_layer or self._rpc_layer, master)
183 def rpc_layer(self): member in GCEClusterResolver
186 @rpc_layer.setter
187 def rpc_layer(self, rpc_layer): member in GCEClusterResolver
188 self._rpc_layer = rpc_layer
|
| D | cluster_resolver_test.py | 38 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
164 rpc_layer="grpc")
170 self.assertEqual(simple_resolver.rpc_layer, "grpc")
181 rpc_layer="grpc")
185 simple_resolver.rpc_layer = "http"
189 self.assertEqual(simple_resolver.rpc_layer, "http")
198 actual_master = simple_resolver.master("worker", 0, rpc_layer="grpc")
208 actual_master = simple_resolver.master("worker", 2, rpc_layer="grpc")
229 rpc_layer="grpc")
238 rpc_layer="http")
[all …]
|
| D | kubernetes_cluster_resolver.py | 47 rpc_layer='grpc', argument
94 self.rpc_layer = rpc_layer
96 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
118 rpc_layer or self.rpc_layer)
|
| D | tpu_cluster_resolver.py | 277 self.rpc_layer = None
279 self.rpc_layer = 'grpc'
287 if self.rpc_layer is not None and tpu.startswith(self.rpc_layer + '://'):
288 tpu = tpu[len(self.rpc_layer + '://'):]
331 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
376 return format_master_url(master, rpc_layer or self.rpc_layer)
434 if self.rpc_layer is None:
443 if (self.rpc_layer is not None and
444 tpu.startswith(self.rpc_layer + '://')):
445 tpus.append(tpu[len(self.rpc_layer + '://'):])
|
| D | slurm_cluster_resolver.py | 60 rpc_layer='grpc'): argument
118 self.rpc_layer = rpc_layer
198 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
216 rpc_layer or self.rpc_layer)
|
| D | tfconfig_cluster_resolver_test.py | 160 self.assertEqual('grpc', cluster_resolver.rpc_layer)
185 cluster_resolver.rpc_layer = 'test'
190 self.assertEqual('test', cluster_resolver.rpc_layer)
|
| D | slurm_cluster_resolver_test.py | 89 slurm_cluster_resolver.rpc_layer = 'ab'
92 slurm_cluster_resolver.master('ps', 0, rpc_layer='test'),
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_coordinator.py | 118 rpc_layer="grpc", argument
143 self._rpc_layer = rpc_layer
332 rpc_layer="", argument
353 rpc_layer=rpc_layer,
386 rpc_layer=None, argument
397 assert _thread_local.rpc_layer == rpc_layer
407 _thread_local.rpc_layer = rpc_layer
412 if rpc_layer:
413 target = rpc_layer + "://" + target
444 protocol=rpc_layer)
[all …]
|
| D | distribute_coordinator_test.py | 395 rpc_layer=None, argument
527 rpc_layer="grpc")
560 rpc_layer=None)
629 rpc_layer=None)
668 rpc_layer=None)
693 rpc_layer=None)
729 rpc_layer=None)
797 rpc_layer=None, argument
800 rpc_layer_from_coordinator[0] = rpc_layer
913 rpc_layer=None): argument
[all …]
|
| D | collective_all_reduce_strategy.py | 145 self._rpc_layer = cluster_resolver.rpc_layer
209 self._rpc_layer = cluster_resolver.rpc_layer
230 protocol=cluster_resolver.rpc_layer or "grpc")
378 rpc_layer=self._rpc_layer)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.cluster_resolver.-simple-cluster-resolver.pbtxt | 11 name: "rpc_layer"
24 …', \'task_type\', \'task_id\', \'environment\', \'num_accelerators\', \'rpc_layer\'], varargs=None…
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| D | tensorflow.distribute.cluster_resolver.-t-f-config-cluster-resolver.pbtxt | 11 name: "rpc_layer"
24 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\', \'environment\'], varargs=Non…
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| D | tensorflow.distribute.cluster_resolver.-g-c-e-cluster-resolver.pbtxt | 11 name: "rpc_layer"
24 …, \'zone\', \'instance_group\', \'port\', \'task_type\', \'task_id\', \'rpc_layer\', \'credentials…
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| D | tensorflow.distribute.cluster_resolver.-kubernetes-cluster-resolver.pbtxt | 12 …argspec: "args=[\'self\', \'job_to_label_mapping\', \'tf_server_port\', \'rpc_layer\', \'override_…
20 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| D | tensorflow.distribute.cluster_resolver.-union-resolver.pbtxt | 11 name: "rpc_layer"
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| D | tensorflow.distribute.cluster_resolver.-slurm-cluster-resolver.pbtxt | 12 …_per_node\', \'gpus_per_task\', \'tasks_per_node\', \'auto_set_gpu\', \'rpc_layer\'], varargs=None…
24 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.cluster_resolver.-g-c-e-cluster-resolver.pbtxt | 11 name: "rpc_layer"
24 …, \'zone\', \'instance_group\', \'port\', \'task_type\', \'task_id\', \'rpc_layer\', \'credentials…
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| D | tensorflow.distribute.cluster_resolver.-t-f-config-cluster-resolver.pbtxt | 11 name: "rpc_layer"
24 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\', \'environment\'], varargs=Non…
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| D | tensorflow.distribute.cluster_resolver.-simple-cluster-resolver.pbtxt | 11 name: "rpc_layer"
24 …', \'task_type\', \'task_id\', \'environment\', \'num_accelerators\', \'rpc_layer\'], varargs=None…
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| D | tensorflow.distribute.cluster_resolver.-kubernetes-cluster-resolver.pbtxt | 12 …argspec: "args=[\'self\', \'job_to_label_mapping\', \'tf_server_port\', \'rpc_layer\', \'override_…
20 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| D | tensorflow.distribute.cluster_resolver.-union-resolver.pbtxt | 11 name: "rpc_layer"
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| D | tensorflow.distribute.cluster_resolver.-slurm-cluster-resolver.pbtxt | 12 …_per_node\', \'gpus_per_task\', \'tasks_per_node\', \'auto_set_gpu\', \'rpc_layer\'], varargs=None…
24 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | collective_all_reduce_strategy.py | 82 rpc_layer=tfconfig.rpc_layer)
|