Home
last modified time | relevance | path

Searched refs:initialize_tpu_system (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy_test.py84 tpu_strategy_util.initialize_tpu_system(resolver)
181 tpu_strategy_util.initialize_tpu_system(resolver)
184 tpu_strategy_util.initialize_tpu_system(resolver)
336 topology = tpu_strategy_util.initialize_tpu_system(resolver)
447 topology = tpu_strategy_util.initialize_tpu_system(resolver)
488 topology = tpu_strategy_util.initialize_tpu_system(resolver)
1268 tpu_strategy_util.initialize_tpu_system(resolver)
1400 topology = tpu_strategy_util.initialize_tpu_system(resolver)
1412 topology = tpu_strategy_util.initialize_tpu_system(resolver)
1425 topology = tpu_strategy_util.initialize_tpu_system(resolver)
[all …]
Dtpu_strategy_compilation_test.py44 tpu_strategy_util.initialize_tpu_system(resolver)
Dstrategy_combinations.py107 _topology = tpu_strategy_util.initialize_tpu_system(resolver)
Dtpu_strategy_model_parallelism_test.py64 topology = tpu_strategy_util.initialize_tpu_system(resolver)
Dcollective_all_reduce_strategy.py536 tpu_strategy_util.initialize_tpu_system()
Dcollective_all_reduce_strategy_test.py81 tpu_strategy_util.initialize_tpu_system()
Dtpu_strategy.py1176 tpu_strategy_util.initialize_tpu_system(self._tpu_cluster_resolver)
Dvars_test.py972 tpu_strategy_util.initialize_tpu_system(resolver)
/external/tensorflow/tensorflow/python/tpu/tests/
Dtpu_initialization_test.py29 tpu_strategy_util.initialize_tpu_system(resolver)
Dtpu_embedding_v2_checkpoint_test.py89 tpu_strategy_util.initialize_tpu_system(self.resolver)
151 tpu_strategy_util.initialize_tpu_system(self.resolver)
Dtpu_embedding_base_test.py157 tpu_strategy_util.initialize_tpu_system(self.resolver)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.tpu.experimental.pbtxt24 name: "initialize_tpu_system"
Dtensorflow.experimental.dtensor.pbtxt84 name: "initialize_tpu_system"
/external/tensorflow/tensorflow/python/eager/
Dremote_cloud_tpu_test.py78 tpu_strategy_util.initialize_tpu_system(resolver)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.tpu.experimental.pbtxt44 name: "initialize_tpu_system"
/external/tensorflow/tensorflow/compiler/tests/
Dconst_op_test.py48 tpu_strategy_util.initialize_tpu_system(resolver)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_strategy_util.py41 def initialize_tpu_system(cluster_resolver=None): function
Dtpu_outside_compilation_test.py73 tpu_strategy_util.initialize_tpu_system(resolver)
/external/tensorflow/tensorflow/python/ops/memory_tests/
Dcustom_gradient_memory_test.py119 tpu_strategy_util.initialize_tpu_system()
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/
Dtpu_cluster_resolver.py110 tpu_strategy_util.initialize_tpu_system(resolver)
/external/tensorflow/tensorflow/python/distribute/integration_test/
Dtpu_memory_test.py73 tf.tpu.experimental.initialize_tpu_system(self.resolver)
/external/tensorflow/tensorflow/python/distribute/parallel_device/
Dparallel_device_test.py100 tpu_strategy_util.initialize_tpu_system()
/external/tensorflow/tensorflow/python/framework/
DBUILD1104 disable_tfrt = True, # TODO(tfrt-devs): Update this test to call initialize_tpu_system
/external/tensorflow/
DRELEASE.md6198 `tf.tpu.experimental.initialize_tpu_system`.