Home
last modified time | relevance | path

Searched refs:cluster_def (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/python/client/
Dsession_clusterspec_prop_test.py50 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 …]
Dsession_list_devices_test.py67 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/
Dtpu_system_metadata.py72 def _query_tpu_system_metadata(master_address, cluster_def=None, argument
98 cluster_def)) as sess:
141 topology = _obtain_topology(master_address, cluster_def)
171 def _obtain_topology(master_address, cluster_def): argument
179 _INITIAL_TPU_SYSTEM_TIMEOUT_IN_MS, cluster_def)
191 def get_session_config_with_timeout(timeout_in_secs, cluster_def): argument
194 operation_timeout_in_ms=timeout_in_secs, cluster_def=cluster_def)
198 def master_job(master, cluster_def): argument
218 if (not cluster_def or not cluster_def.job):
220 job_names = set(job.name for job in cluster_def.job)
[all …]
Dtpu_strategy_util.py121 session_config.cluster_def.CopyFrom(cluster_spec.as_cluster_def())
208 session_config.cluster_def.CopyFrom(cluster_spec.as_cluster_def())
/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy_v2_test.py50 cluster_def = multi_worker_test_base.create_in_process_cluster(
52 cls.cluster_resolver = SimpleClusterResolver(ClusterSpec(cluster_def))
149 cluster_def = multi_worker_test_base.create_in_process_cluster(
151 cls.cluster_resolver = SimpleClusterResolver(ClusterSpec(cluster_def))
428 cluster_def = multi_worker_test_base._create_cluster(
433 cluster_def["chief"] = [
437 ClusterSpec(cluster_def), rpc_layer="grpc")
442 cluster_def = multi_worker_test_base._create_cluster(
444 cluster_def["chief"] = [
448 ClusterSpec(cluster_def), rpc_layer="grpc")
[all …]
Dmulti_worker_util_test.py44 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))
/external/tensorflow/tensorflow/python/training/
Dserver_lib_test.py194 cluster_def = server_lib.ClusterSpec({
198 cluster=cluster_def,
212 cluster_def = cluster_pb2.ClusterDef()
213 job = cluster_def.job.add()
216 job = cluster_def.job.add()
219 return cluster_def
235 cluster_def=get_cluster_def(master_old, worker))
246 cluster_def=get_cluster_def(master_new, worker))
396 cluster_def = server_lib.ClusterSpec({
400 cluster=cluster_def, job_name="local", task_index=0, protocol="grpc")
[all …]
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dmetric_utils_test.py42 cluster_def = multi_worker_test_base.create_in_process_cluster(
44 cluster_def['chief'] = [
48 ClusterSpec(cluster_def), rpc_layer=self.get_rpc_layer())
/external/tensorflow/tensorflow/python/keras/utils/
Ddataset_creator_test.py53 cluster_def = multi_worker_test_base.create_in_process_cluster(
55 cluster_def["chief"] = [
59 SimpleClusterResolver(ClusterSpec(cluster_def), rpc_layer="grpc"))
/external/tensorflow/tensorflow/python/eager/
Dremote.py154 cluster_def = copy.deepcopy(cluster_spec.as_cluster_def())
166 job_def = cluster_def.job.add()
173 cluster=cluster_def,
Dremote_execution_test.py49 cluster_def = cluster_pb2.ClusterDef()
50 job_def = cluster_def.job.add()
58 cluster=cluster_def,
Dremote_cluster_test.py49 cluster_def = cluster_pb2.ClusterDef()
50 job_def = cluster_def.job.add()
58 cluster=cluster_def,
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_env.h38 const ClusterDef* cluster_def = nullptr; member
50 cluster_def = &server_def.cluster(); in WorkerCacheFactoryOptions()
Dmaster.cc378 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()
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/
Dtpu_cluster_resolver.py296 cluster_def = cluster_spec.as_cluster_def() if cluster_spec else None
300 cluster_def=cluster_def,
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_remote_test.cc33 tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); in GetServerDef() local
34 tensorflow::JobDef* job_def = cluster_def->add_job(); in GetServerDef()
/external/tensorflow/tensorflow/python/keras/distribute/
Dparameter_server_training_test.py66 cluster_def = multi_worker_test_base.create_in_process_cluster(
68 cluster_def["chief"] = [
71 return SimpleClusterResolver(ClusterSpec(cluster_def), rpc_layer="grpc")
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dreplicate_test.py228 cluster_def = cluster_pb2.ClusterDef()
229 job_def = cluster_def.job.add()
237 cluster=cluster_def,
/external/tensorflow/tensorflow/python/data/kernel_tests/
Diterator_test.py663 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/
Dtensorflow.-config-proto.pbtxt89 name: "cluster_def"
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_server_lib.cc338 for (const auto& job : options.cluster_def->job()) { in ParseChannelSpec()
365 options.cluster_def->DebugString()); in WorkerCacheFactory()
/external/tensorflow/tensorflow/c/eager/
Dc_api_cluster_test.cc379 tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); in TestRemoteExecuteUpdateServerDefWithFailures() local
380 tensorflow::JobDef* job_def = cluster_def->mutable_job(0); in TestRemoteExecuteUpdateServerDefWithFailures()
Dc_api_test_util.cc364 tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); in GetServerDef() local
365 tensorflow::JobDef* job_def = cluster_def->add_job(); in GetServerDef()
/external/tensorflow/tensorflow/python/framework/
Dconfig_test.py715 cluster_def = cluster_pb2.ClusterDef()
716 job_def = cluster_def.job.add()
721 cluster=cluster_def, job_name=job_name, task_index=0, protocol='grpc')
/external/tensorflow/tensorflow/python/distribute/integration_test/
Dsaved_model_test.py496 cluster_def = multi_worker_test_base.create_in_process_cluster(
499 tf.train.ClusterSpec(cluster_def))

12