| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| 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)
87 rpc_layer=None, argument
102 self._rpc_layer = rpc_layer
134 def rpc_layer(self): member in TFConfigClusterResolver
140 @rpc_layer.setter
141 def rpc_layer(self, rpc_layer): argument
142 self._rpc_layer = rpc_layer
164 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
[all …]
|
| D | cluster_resolver.py | 35 def format_master_url(master, rpc_layer=None): argument
36 if rpc_layer:
37 return '%s://%s' % (rpc_layer, master)
115 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
326 rpc_layer=None): argument
335 self._rpc_layer = rpc_layer
349 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
370 return format_master_url(master, rpc_layer=rpc_layer or self._rpc_layer)
414 def rpc_layer(self): member in SimpleClusterResolver
417 @rpc_layer.setter
[all …]
|
| D | sagemaker_cluster_resolver.py | 44 def format_master_url(master, rpc_layer=None): argument
45 if rpc_layer:
46 return '%s://%s' % (rpc_layer, master)
97 rpc_layer=None, argument
111 self._rpc_layer = rpc_layer
144 def rpc_layer(self): member in SageMakerClusterResolver
150 @rpc_layer.setter
151 def rpc_layer(self, rpc_layer): argument
152 self._rpc_layer = rpc_layer
171 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
[all …]
|
| D | gce_cluster_resolver.py | 75 rpc_layer='grpc', argument
111 self._rpc_layer = rpc_layer
174 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
180 if rpc_layer or self._rpc_layer:
181 return '%s://%s' % (rpc_layer or self._rpc_layer, master)
206 def rpc_layer(self): member in GCEClusterResolver
209 @rpc_layer.setter
210 def rpc_layer(self, rpc_layer): argument
211 self._rpc_layer = rpc_layer
|
| D | cluster_resolver_test.py | 39 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
165 rpc_layer="grpc")
171 self.assertEqual(simple_resolver.rpc_layer, "grpc")
182 rpc_layer="grpc")
186 simple_resolver.rpc_layer = "http"
190 self.assertEqual(simple_resolver.rpc_layer, "http")
199 actual_master = simple_resolver.master("worker", 0, rpc_layer="grpc")
209 actual_master = simple_resolver.master("worker", 2, rpc_layer="grpc")
230 rpc_layer="grpc")
239 rpc_layer="http")
[all …]
|
| D | kubernetes_cluster_resolver.py | 72 rpc_layer='grpc', argument
119 self.rpc_layer = rpc_layer
121 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
145 rpc_layer or self.rpc_layer)
|
| D | slurm_cluster_resolver.py | 186 rpc_layer='grpc'): argument
265 self.rpc_layer = rpc_layer
373 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
391 rpc_layer or self.rpc_layer)
|
| D | sagemaker_cluster_resolver_test.py | 76 cluster_resolver = SageMakerClusterResolver(rpc_layer='grpc')
91 cluster_resolver.rpc_layer = 'test'
96 self.assertEqual('test', cluster_resolver.rpc_layer)
|
| D | slurm_cluster_resolver_test.py | 83 slurm_cluster_resolver.master('worker', 0, rpc_layer='grpc'),
136 slurm_cluster_resolver.rpc_layer = 'ab'
139 slurm_cluster_resolver.master('ps', 0, rpc_layer='test'),
|
| D | tfconfig_cluster_resolver_test.py | 161 self.assertEqual('grpc', cluster_resolver.rpc_layer)
186 cluster_resolver.rpc_layer = 'test'
191 self.assertEqual('test', cluster_resolver.rpc_layer)
|
| /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,
389 rpc_layer=None, argument
400 assert _thread_local.rpc_layer == rpc_layer
410 _thread_local.rpc_layer = rpc_layer
415 if rpc_layer:
416 target = rpc_layer + "://" + target
447 protocol=rpc_layer)
[all …]
|
| D | multi_worker_test_base.py | 160 rpc_layer='grpc'): argument
207 protocol=rpc_layer)
225 self._rpc_layer = cluster_resolver.rpc_layer
235 rpc_layer = cluster_resolver.rpc_layer
240 rpc_layer)
254 protocol=rpc_layer,
288 rpc_layer=self._rpc_layer,
357 rpc_layer='grpc'): argument
366 server_lib.ClusterSpec(cluster_spec), rpc_layer=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.-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.-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.-kubernetes-cluster-resolver.pbtxt | 20 …argspec: "args=[\'self\', \'job_to_label_mapping\', \'tf_server_port\', \'rpc_layer\', \'override_…
28 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| D | tensorflow.distribute.cluster_resolver.-slurm-cluster-resolver.pbtxt | 20 …_per_node\', \'gpus_per_task\', \'tasks_per_node\', \'auto_set_gpu\', \'rpc_layer\'], varargs=None…
32 …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.-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.-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.-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.-kubernetes-cluster-resolver.pbtxt | 20 …argspec: "args=[\'self\', \'job_to_label_mapping\', \'tf_server_port\', \'rpc_layer\', \'override_…
28 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| D | tensorflow.distribute.cluster_resolver.-slurm-cluster-resolver.pbtxt | 20 …_per_node\', \'gpus_per_task\', \'tasks_per_node\', \'auto_set_gpu\', \'rpc_layer\'], varargs=None…
32 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'rpc_layer\'], varargs=None, keywords=None,…
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | metric_utils_test.py | 43 num_workers=1, num_ps=1, rpc_layer=self.get_rpc_layer())
48 ClusterSpec(cluster_def), rpc_layer=self.get_rpc_layer())
|