| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| H A D | cluster_resolver.py | 31 def format_master_url(master, rpc_layer=None): argument
111 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
322 rpc_layer=None): argument
345 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
410 def rpc_layer(self): member in SimpleClusterResolver
414 def rpc_layer(self, rpc_layer): member in SimpleClusterResolver
569 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
619 def rpc_layer(self): member in UnionClusterResolver
623 def rpc_layer(self, rpc_layer): argument
|
| H A D | tfconfig_cluster_resolver.py | 31 def format_master_url(master, rpc_layer=None): argument
83 rpc_layer=None, argument
130 def rpc_layer(self): member in TFConfigClusterResolver
137 def rpc_layer(self, rpc_layer): argument
160 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
|
| H A D | sagemaker_cluster_resolver.py | 40 def format_master_url(master, rpc_layer=None): argument
93 rpc_layer=None, argument
140 def rpc_layer(self): member in SageMakerClusterResolver
147 def rpc_layer(self, rpc_layer): member in SageMakerClusterResolver
167 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
|
| H A D | gce_cluster_resolver.py | 71 rpc_layer='grpc', argument
170 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
202 def rpc_layer(self): member in GCEClusterResolver
206 def rpc_layer(self, rpc_layer): member in GCEClusterResolver
|
| H A D | kubernetes_cluster_resolver.py | 61 rpc_layer='grpc', argument
112 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
|
| H A D | slurm_cluster_resolver.py | 182 rpc_layer='grpc'): argument
369 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
|
| H A D | cluster_resolver_test.py | 35 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | distribute_coordinator.py | 114 rpc_layer="grpc", argument
328 rpc_layer="", argument
385 rpc_layer=None, argument
451 cluster_spec, session_config, rpc_layer): argument
494 cluster_spec, session_config, rpc_layer): argument
636 rpc_layer="grpc"): argument
|
| H A D | distribute_coordinator_test.py | 400 rpc_layer=None, argument
791 rpc_layer=None, argument
907 rpc_layer=None): argument
|
| H A D | multi_process_runner.py | 126 rpc_layer=None, argument
1141 def _set_tf_config(task_type, task_id, cluster_spec, rpc_layer=None): argument
1158 rpc_layer=None, argument
|
| H A D | multi_worker_test_base.py | 127 rpc_layer='grpc'): argument
332 rpc_layer='grpc', argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/ |
| H A D | distribute_coordinator_utils.py | 79 rpc_layer="grpc", argument
293 rpc_layer="", argument
349 rpc_layer=None, argument
452 rpc_layer="grpc"): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/experimental/rpc/ |
| H A D | rpc_ops.py | 58 def create(rpc_layer, address): argument
122 def create(rpc_layer, address, name="", timeout_in_ms=0): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/ |
| H A D | tpu_cluster_resolver.py | 229 def master(self, task_type=None, task_id=None, rpc_layer=None): argument
|