Searched refs:TPUClusterResolver (Results 1 – 18 of 18) sorted by relevance
28 from tensorflow.python.distribute.cluster_resolver import TPUClusterResolver134 @mock.patch.object(TPUClusterResolver,139 TPUClusterResolver(tpu='')145 self.assertTrue(TPUClusterResolver._isRunningInGCE())151 self.assertFalse(TPUClusterResolver._isRunningInGCE())153 @mock.patch.object(TPUClusterResolver,165 resolver = TPUClusterResolver(187 @mock.patch.object(TPUClusterResolver,199 resolver = TPUClusterResolver(214 @mock.patch.object(TPUClusterResolver,[all …]
60 class TPUClusterResolver(ClusterResolver): class480 device_details = TPUClusterResolver._get_device_dict_and_cores(495 return {'TPU': TPUClusterResolver._verify_and_return_same_core_count(
28 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
1 path: "tensorflow.distribute.cluster_resolver.TPUClusterResolver"3 …"<class \'tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver.TPUClusterResolver\'>"
28 name: "TPUClusterResolver"
23 from tensorflow.python.distribute.cluster_resolver import TPUClusterResolver67 cluster_resolver = TPUClusterResolver("")
26 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
32 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
30 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
104 tf.contrib.cluster_resolver.TPUClusterResolver(
88 cluster_resolver = tf.distribute.cluster_resolver.TPUClusterResolver(
31 from tensorflow.python.distribute.cluster_resolver import TPUClusterResolver203 tpu_cluster_resolver = TPUClusterResolver("")
193 tpu_cluster_resolver = tf.contrib.cluster_resolver.TPUClusterResolver(
332 resolver = tpu_cluster_resolver.TPUClusterResolver("")
221 tpu_cluster_resolver = tpu_cluster_resolver_lib.TPUClusterResolver('')
574 * Integrate `TPUClusterResolver` with GKE's integration for Cloud TPUs.