Home
last modified time | relevance | path

Searched refs:create_in_process_cluster (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dcollective_all_reduce_strategy_test.py293 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
431 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
460 cluster_resolver=multi_worker_test_base.create_in_process_cluster(
H A Dparameter_server_strategy_test.py572 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
740 cluster_spec = multi_worker_test_base.create_in_process_cluster(
825 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
H A Dmulti_worker_test_base.py123 def create_in_process_cluster(num_workers, function
434 cls._cluster_spec = create_in_process_cluster(num_workers=num_workers,
H A Dtest_util.py57 cluster_params.cluster = multi_worker_test_base.create_in_process_cluster(
H A Dstrategy_combinations.py227 cluster_def = multi_worker_test_base.create_in_process_cluster(
H A Dmirrored_strategy_test.py1317 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/
H A Dmetric_utils_test.py37 cluster_def = multi_worker_test_base.create_in_process_cluster(
H A Dcluster_coordinator_test.py486 cluster_def = multi_worker_test_base.create_in_process_cluster(
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/integration_test/
H A Dsaved_model_test.py564 cluster_def = multi_worker_test_base.create_in_process_cluster(
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/
H A Dcross_device_ops_test.py443 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(