Home
last modified time | relevance | path

Searched refs:ClusterSpec (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dcluster_resolver_test.py130 expected_proto, server_lib.ClusterSpec(cluster_spec).as_cluster_def())
133 server_lib.ClusterSpec(cluster_spec.as_cluster_def()).as_cluster_def())
136 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
139 base_cluster_spec = server_lib.ClusterSpec({
157 base_cluster_spec = server_lib.ClusterSpec({
174 base_cluster_spec = server_lib.ClusterSpec({
193 base_cluster_spec = server_lib.ClusterSpec({
203 base_cluster_spec = server_lib.ClusterSpec({
213 base_cluster_spec = server_lib.ClusterSpec({
223 cluster_spec_1 = server_lib.ClusterSpec({
[all …]
DREADME.md3 …ClusterSpec` framework, Cluster Resolvers allow users to simply specify a configuration and a clus…
5 `ClusterResolvers` are designed to work well with `ManagedTrainingSession` and `ClusterSpec` propag…
Dtfconfig_cluster_resolver.py26 from tensorflow.python.training.server_lib import ClusterSpec
161 return ClusterSpec({})
162 return ClusterSpec(tf_config['cluster'])
Dsagemaker_cluster_resolver_test.py38 server_lib.ClusterSpec(cluster_spec).as_cluster_def())
41 server_lib.ClusterSpec(cluster_spec.as_cluster_def()).as_cluster_def())
44 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
Dkubernetes_cluster_resolver_test.py67 server_lib.ClusterSpec(cluster_spec).as_cluster_def())
69 server_lib.ClusterSpec(
72 server_lib.ClusterSpec(
Dslurm_cluster_resolver_test.py56 server_lib.ClusterSpec(cluster_spec).as_cluster_def())
59 server_lib.ClusterSpec(cluster_spec.as_cluster_def()).as_cluster_def())
62 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
Dsagemaker_cluster_resolver.py25 from tensorflow.python.training.server_lib import ClusterSpec
168 return ClusterSpec({})
169 return ClusterSpec(tf_config['cluster'])
Dtfconfig_cluster_resolver_test.py40 expected_proto, server_lib.ClusterSpec(cluster_spec).as_cluster_def())
43 server_lib.ClusterSpec(cluster_spec.as_cluster_def()).as_cluster_def())
46 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
Dgce_cluster_resolver_test.py35 expected_proto, server_lib.ClusterSpec(cluster_spec).as_cluster_def())
38 server_lib.ClusterSpec(cluster_spec.as_cluster_def()).as_cluster_def())
41 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
Dgce_cluster_resolver.py22 from tensorflow.python.training.server_lib import ClusterSpec
172 return ClusterSpec({self._task_type: worker_list})
Dslurm_cluster_resolver.py27 from tensorflow.python.training.server_lib import ClusterSpec
357 return ClusterSpec(self._cluster_allocation)
/external/tensorflow/tensorflow/python/training/
Dserver_lib_test.py194 cluster_def = server_lib.ClusterSpec({
396 cluster_def = server_lib.ClusterSpec({
410 cluster_spec = server_lib.ClusterSpec(cluster_def)
414 cluster_def = server_lib.ClusterSpec({
429 cluster_spec = server_lib.ClusterSpec(cluster_def)
433 cluster_def = server_lib.ClusterSpec({
452 cluster_spec = server_lib.ClusterSpec(cluster_def)
456 cluster_def = server_lib.ClusterSpec({
477 cluster_spec = server_lib.ClusterSpec(cluster_def)
484 cluster_spec = server_lib.ClusterSpec({
[all …]
Ddevice_setter.py203 if isinstance(cluster, server_lib.ClusterSpec):
206 cluster_spec = server_lib.ClusterSpec(cluster).as_dict()
Ddevice_setter_test.py32 _cluster_spec = server_lib.ClusterSpec({
149 cluster_spec = server_lib.ClusterSpec({
170 cluster_spec = server_lib.ClusterSpec({
Dserver_lib.py70 cluster_spec = ClusterSpec(server_or_cluster_def)
247 class ClusterSpec(object): class
309 elif isinstance(cluster, ClusterSpec):
/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy_v2_test.py40 from tensorflow.python.training.server_lib import ClusterSpec
52 cls.cluster_resolver = SimpleClusterResolver(ClusterSpec(cluster_def))
151 cls.cluster_resolver = SimpleClusterResolver(ClusterSpec(cluster_def))
437 ClusterSpec(cluster_def), rpc_layer="grpc")
448 ClusterSpec(cluster_def), rpc_layer="grpc")
459 ClusterSpec(cluster_def), rpc_layer="grpc")
470 ClusterSpec(cluster_def), rpc_layer="grpc", task_type="foobar")
480 ClusterSpec(cluster_def),
495 ClusterSpec(cluster_def), rpc_layer="grpc", task_type="ps", task_id=0)
507 ClusterSpec(cluster_def),
Dmulti_worker_util_test.py31 server_lib.ClusterSpec(lhs).as_dict(),
32 server_lib.ClusterSpec(rhs).as_dict())
63 cluster_spec = server_lib.ClusterSpec({
Dmulti_worker_util.py41 return server_lib.ClusterSpec(cluster_spec)
42 elif not isinstance(cluster_spec, server_lib.ClusterSpec):
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.train.-cluster-spec.pbtxt1 path: "tensorflow.train.ClusterSpec"
3 is_instance: "<class \'tensorflow.python.training.server_lib.ClusterSpec\'>"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.-cluster-spec.pbtxt1 path: "tensorflow.train.ClusterSpec"
3 is_instance: "<class \'tensorflow.python.training.server_lib.ClusterSpec\'>"
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dmetric_utils_test.py30 from tensorflow.python.training.server_lib import ClusterSpec
48 ClusterSpec(cluster_def), rpc_layer=self.get_rpc_layer())
/external/tensorflow/tensorflow/python/keras/utils/
Ddataset_creator_test.py30 from tensorflow.python.training.server_lib import ClusterSpec
59 SimpleClusterResolver(ClusterSpec(cluster_def), rpc_layer="grpc"))
/external/tensorflow/tensorflow/python/eager/
Dremote.py74 cluster_spec = server_lib.ClusterSpec(
142 if isinstance(cluster_spec_or_resolver, server_lib.ClusterSpec):
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/
Dtpu_cluster_resolver_test.py117 server_lib.ClusterSpec(cluster_spec).as_cluster_def())
120 server_lib.ClusterSpec(cluster_spec.as_cluster_def()).as_cluster_def())
123 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
455 server_lib.ClusterSpec({
Dtpu_cluster_resolver.py340 return server_lib.ClusterSpec(cluster_spec)
342 return server_lib.ClusterSpec({})

12