Home
last modified time | relevance | path

Searched refs:as_cluster_def (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dbrain_jobs_cluster_resolver_test.py32 self.assertProtoEquals(expected_proto, cluster_spec.as_cluster_def())
34 expected_proto, server_lib.ClusterSpec(cluster_spec).as_cluster_def())
37 server_lib.ClusterSpec(cluster_spec.as_cluster_def()).as_cluster_def())
40 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
Dkubernetes_cluster_resolver_test.py64 self.assertProtoEquals(expected_proto, cluster_spec.as_cluster_def())
67 server_lib.ClusterSpec(cluster_spec).as_cluster_def())
70 cluster_spec.as_cluster_def()).as_cluster_def())
73 cluster_spec.as_dict()).as_cluster_def())
Dslurm_cluster_resolver_test.py36 self.assertProtoEquals(expected_proto, cluster_spec.as_cluster_def())
39 server_lib.ClusterSpec(cluster_spec).as_cluster_def())
42 server_lib.ClusterSpec(cluster_spec.as_cluster_def()).as_cluster_def())
45 server_lib.ClusterSpec(cluster_spec.as_dict()).as_cluster_def())
Dtfconfig_cluster_resolver_test.py38 self.assertProtoEquals(expected_proto, cluster_spec.as_cluster_def())
40 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.py33 self.assertProtoEquals(expected_proto, cluster_spec.as_cluster_def())
35 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())
Dcluster_resolver_test.py128 self.assertProtoEquals(expected_proto, cluster_spec.as_cluster_def())
130 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())
Dtpu_cluster_resolver_test.py114 self.assertProtoEquals(expected_proto, cluster_spec.as_cluster_def())
117 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())
/external/tensorflow/tensorflow/python/training/
Dserver_lib_test.py197 }).as_cluster_def()
346 }).as_cluster_def()
359 self.assertProtoEquals(cluster_def, cluster_spec.as_cluster_def())
364 }).as_cluster_def()
378 self.assertProtoEquals(cluster_def, cluster_spec.as_cluster_def())
384 }).as_cluster_def()
401 self.assertProtoEquals(cluster_def, cluster_spec.as_cluster_def())
410 }).as_cluster_def()
426 self.assertProtoEquals(cluster_def, cluster_spec.as_cluster_def())
456 self.assertProtoEquals(expected_proto, cluster_spec.as_cluster_def())
[all …]
Ddevice_setter_test.py137 cluster=self._cluster_spec.as_cluster_def())):
158 cluster=cluster_spec.as_cluster_def())):
179 cluster=cluster_spec.as_cluster_def())):
Dserver_lib.py89 cluster=cluster_spec.as_cluster_def(),
306 self._cluster_def.MergeFrom(cluster.as_cluster_def())
363 def as_cluster_def(self): member in ClusterSpec
/external/tensorflow/tensorflow/python/tpu/
Dtpu_strategy_util.py114 session_config.cluster_def.CopyFrom(cluster_spec.as_cluster_def())
195 session_config.cluster_def.CopyFrom(cluster_spec.as_cluster_def())
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.train.-cluster-spec.pbtxt14 name: "as_cluster_def"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.-cluster-spec.pbtxt14 name: "as_cluster_def"
/external/tensorflow/tensorflow/python/eager/
Dremote.py155 cluster_def = copy.deepcopy(cluster_spec.as_cluster_def())
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy.py67 cluster_def = cluster_spec.as_cluster_def() if cluster_spec else None
738 updated_config.cluster_def.CopyFrom(cluster_spec.as_cluster_def())
Dcollective_all_reduce_strategy.py291 cluster=cluster_spec.as_cluster_def(),