| /external/tensorflow/tensorflow/python/client/ |
| D | session_clusterspec_prop_test.py | 50 cluster_def = cluster_pb2.ClusterDef()
51 job = cluster_def.job.add()
55 config = config_pb2.ConfigProto(cluster_def=cluster_def)
65 cluster_def = cluster_pb2.ClusterDef()
66 job = cluster_def.job.add()
70 config = config_pb2.ConfigProto(cluster_def=cluster_def)
93 cluster_def = cluster_pb2.ClusterDef()
94 job = cluster_def.job.add()
98 config = config_pb2.ConfigProto(cluster_def=cluster_def)
109 cluster_def = cluster_pb2.ClusterDef()
[all …]
|
| D | session_list_devices_test.py | 67 cluster_def = cluster_pb2.ClusterDef()
68 job = cluster_def.job.add()
72 config = config_pb2.ConfigProto(cluster_def=cluster_def)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_system_metadata.py | 53 def _query_tpu_system_metadata(master_address, cluster_def=None, argument
79 cluster_def)) as sess:
122 topology = _obtain_topology(master_address, cluster_def)
152 def _obtain_topology(master_address, cluster_def): argument
160 _INITIAL_TPU_SYSTEM_TIMEOUT_IN_MS, cluster_def)
172 def get_session_config_with_timeout(timeout_in_secs, cluster_def): argument
175 operation_timeout_in_ms=timeout_in_secs, cluster_def=cluster_def)
179 def master_job(master, cluster_def): argument
200 if (not cluster_def or not cluster_def.job):
202 job_names = set(job.name for job in cluster_def.job)
[all …]
|
| D | tpu_strategy_util.py | 114 session_config.cluster_def.CopyFrom(cluster_spec.as_cluster_def())
195 session_config.cluster_def.CopyFrom(cluster_spec.as_cluster_def())
|
| D | tpu_embedding.py | 586 cluster_def=None, argument
642 if master is None and cluster_def is None:
661 cluster_def=cluster_def))
669 cluster_def)
|
| D | datasets_test.py | 56 session_config = config_pb2.ConfigProto(cluster_def=self._cluster_def)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | server_lib_test.py | 195 cluster_def = server_lib.ClusterSpec({
199 cluster=cluster_def,
344 cluster_def = server_lib.ClusterSpec({
348 cluster=cluster_def, job_name="local", task_index=0, protocol="grpc")
358 cluster_spec = server_lib.ClusterSpec(cluster_def)
359 self.assertProtoEquals(cluster_def, cluster_spec.as_cluster_def())
362 cluster_def = server_lib.ClusterSpec({
366 cluster=cluster_def, job_name="local", task_index=1, protocol="grpc")
377 cluster_spec = server_lib.ClusterSpec(cluster_def)
378 self.assertProtoEquals(cluster_def, cluster_spec.as_cluster_def())
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | multi_worker_util_test.py | 44 cluster_def = cluster_pb2.ClusterDef()
45 job = cluster_def.job.add()
49 job = cluster_def.job.add()
54 job = cluster_def.job.add()
60 cluster_def, multi_worker_util.normalize_cluster_spec(cluster_def))
|
| D | tpu_strategy.py | 67 cluster_def = cluster_spec.as_cluster_def() if cluster_spec else None
71 cluster_def=cluster_def,
738 updated_config.cluster_def.CopyFrom(cluster_spec.as_cluster_def())
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | remote.py | 155 cluster_def = copy.deepcopy(cluster_spec.as_cluster_def())
167 job_def = cluster_def.job.add()
174 cluster=cluster_def,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_env.h | 38 const ClusterDef* cluster_def = nullptr; member
50 cluster_def = &server_def.cluster(); in WorkerCacheFactoryOptions()
|
| D | master.cc | 378 worker_cache_factory_options.cluster_def = &req->config().cluster_def(); in CreateSession()
390 for (auto&& job : req->config().cluster_def().job()) { in CreateSession()
398 req->config().cluster_def().ShortDebugString()); in CreateSession()
|
| D | master_session.cc | 1342 if (options.cluster_def) { in CreateWorkerSessions()
1344 *options.cluster_def; in CreateWorkerSessions()
|
| /external/tensorflow/tensorflow/c/experimental/ |
| D | network_test.cc | 135 tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); in GetServerDef() local
136 tensorflow::JobDef* job_def = cluster_def->add_job(); in GetServerDef()
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | replicate_test.py | 190 cluster_def = cluster_pb2.ClusterDef()
191 job_def = cluster_def.job.add()
199 cluster=cluster_def,
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | iterator_test.py | 663 cluster_def = cluster_pb2.ClusterDef()
664 workers = cluster_def.job.add()
668 client = cluster_def.job.add()
671 config = config_pb2.ConfigProto(cluster_def=cluster_def)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-config-proto.pbtxt | 89 name: "cluster_def"
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 602 cluster_def = cluster_pb2.ClusterDef()
603 job_def = cluster_def.job.add()
608 cluster=cluster_def, job_name=job_name, task_index=0, protocol='grpc')
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 317 for (const auto& job : options.cluster_def->job()) { in ParseChannelSpec()
344 options.cluster_def->DebugString()); in WorkerCacheFactory()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_remote_test.cc | 35 tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); in GetServerDef() local
36 tensorflow::JobDef* job_def = cluster_def->add_job(); in GetServerDef()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | config.proto | 463 ClusterDef cluster_def = 14; field
|