| /external/tensorflow/tensorflow/python/client/ |
| D | session_clusterspec_prop_test.py | 56 cluster_def = cluster_pb2.ClusterDef()
57 job = cluster_def.job.add()
61 config = config_pb2.ConfigProto(cluster_def=cluster_def)
71 cluster_def = cluster_pb2.ClusterDef()
72 job = cluster_def.job.add()
76 config = config_pb2.ConfigProto(cluster_def=cluster_def)
99 cluster_def = cluster_pb2.ClusterDef()
100 job = cluster_def.job.add()
104 config = config_pb2.ConfigProto(cluster_def=cluster_def)
115 cluster_def = cluster_pb2.ClusterDef()
[all …]
|
| D | session_list_devices_test.py | 66 cluster_def = cluster_pb2.ClusterDef()
67 job = cluster_def.job.add()
71 config = config_pb2.ConfigProto(cluster_def=cluster_def)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_system_metadata.py | 55 def _query_tpu_system_metadata(master_address, cluster_def=None, argument
85 cluster_def)) as sess:
130 topology = _obtain_topology(master_address, cluster_def)
160 def _obtain_topology(master_address, cluster_def): argument
168 _INITIAL_TPU_SYSTEM_TIMEOUT_IN_MS, cluster_def)
180 def get_session_config_with_timeout(timeout_in_secs, cluster_def): argument
183 operation_timeout_in_ms=timeout_in_secs, cluster_def=cluster_def)
187 def master_job(master, cluster_def): argument
208 if (not cluster_def or not cluster_def.job):
210 job_names = set([job.name for job in cluster_def.job])
[all …]
|
| D | tpu_context.py | 264 cluster_def = None
266 self._config.session_config.cluster_def.job):
267 cluster_def = self._config.session_config.cluster_def
273 cluster_def=cluster_def,
516 cluster_def = (run_config.session_config.cluster_def
519 return tpu_system_metadata_lib.master_job(master, cluster_def)
|
| D | _tpu_estimator_embedding.py | 252 cluster_def = cluster_spec.as_cluster_def() if cluster_spec else None
257 cluster_def = None
265 cluster_def,
|
| D | tpu_config.py | 250 self._session_config.cluster_def.CopyFrom(
|
| D | tpu_embedding.py | 298 cluster_def=None, argument
338 self._cluster_def = cluster_def
341 self._master, cluster_def=self._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 | 59 cluster_def = cluster_spec.as_cluster_def() if cluster_spec else None
63 cluster_def=cluster_def,
608 updated_config.cluster_def.CopyFrom(cluster_spec.as_cluster_def())
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | remote.py | 66 cluster_def = ClusterDef()
67 job_def = cluster_def.job.add()
73 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 | 1281 if (options.cluster_def) { in CreateWorkerSessions()
1283 *options.cluster_def; in CreateWorkerSessions()
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| D | remote_test.py | 63 cluster_def = cluster_pb2.ClusterDef()
64 job_def = cluster_def.job.add()
72 cluster=cluster_def,
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | iterator_test.py | 660 cluster_def = cluster_pb2.ClusterDef()
661 workers = cluster_def.job.add()
665 client = cluster_def.job.add()
668 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/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 284 for (const auto& job : options.cluster_def->job()) { in ParseChannelSpec()
311 options.cluster_def->DebugString()); in WorkerCacheFactory()
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_support.py | 159 config.cluster_def.CopyFrom(cluster_spec.as_cluster_def())
183 cluster_def = cluster_spec.as_cluster_def() if cluster_spec else None
186 master, cluster_def=cluster_def, query_topology=False))
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 118 tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); in GetServerDef() local
119 tensorflow::JobDef* job_def = cluster_def->add_job(); in GetServerDef()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | config.proto | 406 ClusterDef cluster_def = 14; field
|