Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/training/
Dserver_lib_test.py196 }).as_cluster_def()
398 }).as_cluster_def()
411 self.assertProtoEquals(cluster_def, cluster_spec.as_cluster_def())
416 }).as_cluster_def()
430 self.assertProtoEquals(cluster_def, cluster_spec.as_cluster_def())
436 }).as_cluster_def()
453 self.assertProtoEquals(cluster_def, cluster_spec.as_cluster_def())
462 }).as_cluster_def()
478 self.assertProtoEquals(cluster_def, cluster_spec.as_cluster_def())
508 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(),
311 self._cluster_def.MergeFrom(cluster.as_cluster_def())
368 def as_cluster_def(self): member in ClusterSpec
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dsagemaker_cluster_resolver_test.py35 self.assertProtoEquals(expected_proto, cluster_spec.as_cluster_def())
38 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.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.py53 self.assertProtoEquals(expected_proto, cluster_spec.as_cluster_def())
56 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())
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())
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/
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())
Dtpu_cluster_resolver.py296 cluster_def = cluster_spec.as_cluster_def() if cluster_spec else None
/external/tensorflow/tensorflow/python/tpu/
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/coordinator/
Dutils.py38 cluster_resolver.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/distribute/
Ddevice_util_test.py88 cluster=cluster_spec.as_cluster_def(),
Dcollective_all_reduce_strategy.py456 cluster=cluster_spec.as_cluster_def(),
Dcross_device_ops_test.py104 cluster=cluster_resolver.cluster_spec().as_cluster_def(),
Dtpu_strategy.py1413 updated_config.cluster_def.CopyFrom(cluster_spec.as_cluster_def())
Dtpu_strategy_test.py985 self.assertProtoEquals(cluster_spec.as_cluster_def(),
/external/tensorflow/tensorflow/python/eager/
Dremote.py154 cluster_def = copy.deepcopy(cluster_spec.as_cluster_def())
/external/tensorflow/tensorflow/python/kernel_tests/
Dcollective_ops_multi_worker_test.py47 cluster=cluster_resolver.cluster_spec().as_cluster_def(),