Searched refs:tpu_cluster_resolver (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tpu/profiler/pip_package/cloud_tpu_profiler/ |
| D | main.py | 96 tpu_cluster_resolver = None
103 tpu_cluster_resolver = (
106 service_addr = tpu_cluster_resolver.get_master()
116 elif tpu_cluster_resolver is not None:
117 workers_list = get_workers_list(tpu_cluster_resolver)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 53 def get_tpu_system_metadata(tpu_cluster_resolver): argument
55 master = tpu_cluster_resolver.master()
58 cluster_spec = tpu_cluster_resolver.cluster_spec()
129 tpu_cluster_resolver=None, argument
147 self, tpu_cluster_resolver, steps_per_run, device_assignment))
197 tpu_cluster_resolver=None, argument
202 if tpu_cluster_resolver is None:
203 tpu_cluster_resolver = TPUClusterResolver("")
210 self._tpu_cluster_resolver = tpu_cluster_resolver
|
| /external/tensorflow/tensorflow/contrib/cluster_resolver/python/training/ |
| D | tpu_cluster_resolver.py | 26 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
|
| D | __init__.py | 32 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
|
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | __init__.py | 28 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
|
| D | BUILD | 62 srcs = ["tpu_cluster_resolver.py"],
|
| /external/tensorflow/tensorflow/contrib/cluster_resolver/ |
| D | __init__.py | 30 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt | 3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver.TPUClust…
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 20 …argspec: "args=[\'self\', \'tpu_cluster_resolver\', \'steps_per_run\', \'device_assignment\'], var…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt | 3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver.TPUClust…
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 20 …argspec: "args=[\'self\', \'tpu_cluster_resolver\', \'steps_per_run\', \'device_assignment\'], var…
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_support.py | 57 from tensorflow.contrib.cluster_resolver.python.training import tpu_cluster_resolver as tpu_cluster…
177 def get_tpu_system_metadata(tpu_cluster_resolver): argument
179 master = tpu_cluster_resolver.master()
182 cluster_spec = tpu_cluster_resolver.cluster_spec()
194 def __init__(self, tpu_cluster_resolver=None, using_single_core=False): argument
220 if tpu_cluster_resolver is None:
221 tpu_cluster_resolver = tpu_cluster_resolver_lib.TPUClusterResolver('')
223 metadata = get_tpu_system_metadata(tpu_cluster_resolver)
225 self._tpu_cluster_resolver = tpu_cluster_resolver
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
| D | main_estimator_tpu.py | 193 tpu_cluster_resolver = tf.contrib.cluster_resolver.TPUClusterResolver(
198 cluster=tpu_cluster_resolver,
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | combinations.py | 56 from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
332 resolver = tpu_cluster_resolver.TPUClusterResolver("")
|