Searched refs:TPUClusterResolver (Results 1 – 24 of 24) sorted by relevance
145 resolver.TPUClusterResolver(tpu='')164 cluster_resolver = resolver.TPUClusterResolver(198 cluster_resolver = resolver.TPUClusterResolver(224 cluster_resolver = resolver.TPUClusterResolver(245 cluster_resolver = resolver.TPUClusterResolver(271 cluster_resolver = resolver.TPUClusterResolver(297 cluster_resolver = resolver.TPUClusterResolver(343 cluster_resolver = resolver.TPUClusterResolver(393 cluster_resolver = resolver.TPUClusterResolver(418 resolver.TPUClusterResolver([all …]
59 class TPUClusterResolver(cluster_resolver.ClusterResolver): class109 resolver = TPUClusterResolver(tpu, zone, project)377 device_details = TPUClusterResolver._get_device_dict_and_cores(395 TPUClusterResolver._verify_and_return_same_core_count(
23 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver64 cluster_resolver = TPUClusterResolver("")65 assert isinstance(cluster_resolver, TPUClusterResolver)167 cluster_resolver = TPUClusterResolver("")168 assert isinstance(cluster_resolver, TPUClusterResolver)
100 resolver = tpu_cluster_resolver.TPUClusterResolver(149 resolver = tpu_cluster_resolver.TPUClusterResolver(
128 self.resolver = tpu_cluster_resolver.TPUClusterResolver(
74 self.resolver = tpu_cluster_resolver.TPUClusterResolver(506 self.resolver = tpu_cluster_resolver.TPUClusterResolver(
64 resolver = tpu_cluster_resolver.TPUClusterResolver(
23 …om tensorflow.python.distribute.cluster_resolver.tpu.tpu_cluster_resolver import TPUClusterResolver26 tf_export('distribute.cluster_resolver.TPUClusterResolver')(TPUClusterResolver)
35 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
1 path: "tensorflow.distribute.cluster_resolver.TPUClusterResolver"3 …ass \'tensorflow.python.distribute.cluster_resolver.tpu.tpu_cluster_resolver.TPUClusterResolver\'>"
28 name: "TPUClusterResolver"
37 resolver = tpu_cluster_resolver.TPUClusterResolver(
66 resolver = tpu_cluster_resolver.TPUClusterResolver(
85 resolver = tpu_cluster_resolver.TPUClusterResolver()91 resolver = tpu_cluster_resolver.TPUClusterResolver(
44 from tensorflow.python.distribute.cluster_resolver import TPUClusterResolver794 tpu_cluster_resolver = TPUClusterResolver("")
72 resolver = tpu_cluster_resolver.TPUClusterResolver(
912 resolver = tpu_cluster_resolver.TPUClusterResolver("")
160 resolver.TPUClusterResolver([FLAGS.tpu],
42 resolver = tf.distribute.cluster_resolver.TPUClusterResolver(
63 resolver = tpu_cluster_resolver.TPUClusterResolver(
1134 …* Add a `tf.distribute.cluster_resolver.TPUClusterResolver.connect` API to simplify TPU initializa…3247 * Integrate `TPUClusterResolver` with GKE's integration for Cloud TPUs.