Home
last modified time | relevance | path

Searched refs:MultiProcessPoolRunner (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dmulti_process_runner_test.py540 _global_pool = multi_process_runner.MultiProcessPoolRunner(
548 runner = multi_process_runner.MultiProcessPoolRunner(cluster_spec)
555 runner = multi_process_runner.MultiProcessPoolRunner(cluster_spec)
564 runner = multi_process_runner.MultiProcessPoolRunner(cluster_spec)
581 runner = multi_process_runner.MultiProcessPoolRunner(cluster_spec)
587 runner = multi_process_runner.MultiProcessPoolRunner(
H A Dcross_device_ops_test.py111 class MultiProcessPoolRunner(): class
116 self.runner = multi_process_runner.MultiProcessPoolRunner(cluster_spec_dict)
127 global_mpr_2p = MultiProcessPoolRunner(num_processes=2)
128 global_mpr_1p = MultiProcessPoolRunner(num_processes=1)
H A Dstrategy_combinations.py311 runner = multi_process_runner.MultiProcessPoolRunner(
H A Dmulti_process_runner.py890 class MultiProcessPoolRunner(object): class
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/
H A Dcollective_ops_multi_worker_test.py155 two_worker_pool_runner = multi_process_runner.MultiProcessPoolRunner(