Home
last modified time | relevance | path

Searched refs:create_cluster_spec (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dmulti_process_runner_test.py83 multi_worker_test_base.create_cluster_spec(
97 multi_worker_test_base.create_cluster_spec(num_workers=1, num_ps=1),
104 cluster_spec = multi_worker_test_base.create_cluster_spec(num_workers=2)
117 multi_worker_test_base.create_cluster_spec(num_workers=1),
134 multi_worker_test_base.create_cluster_spec(num_workers=2),
154 multi_worker_test_base.create_cluster_spec(num_workers=2),
180 multi_worker_test_base.create_cluster_spec(num_workers=2),
212 multi_worker_test_base.create_cluster_spec(
260 multi_worker_test_base.create_cluster_spec(
282 multi_worker_test_base.create_cluster_spec(num_workers=2),
[all …]
Dmulti_worker_continuous_run_test.py102 cluster_spec=test_base.create_cluster_spec(num_workers=NUM_WORKERS))
135 cluster_spec=test_base.create_cluster_spec(num_workers=NUM_WORKERS))
Dmulti_process_runner_no_init_test.py37 multi_worker_test_base.create_cluster_spec(num_workers=1))
Ddevice_util_test.py85 multi_worker_test_base.create_cluster_spec(
Dmulti_worker_test_base.py358 cluster_spec = create_cluster_spec(
373 def create_cluster_spec(has_chief=False, function
Dstrategy_combinations.py189 cluster_spec = multi_worker_test_base.create_cluster_spec(
Dcombinations.py519 cluster_spec = multi_worker_test_base.create_cluster_spec(
Dcollective_all_reduce_strategy_test.py530 cluster_spec = multi_worker_test_base.create_cluster_spec(
Dcross_device_ops_test.py118 cluster_spec_dict = multi_worker_test_base.create_cluster_spec(
/external/tensorflow/tensorflow/python/distribute/integration_test/
Dmwms_peer_failure_test.py89 cluster_spec = multi_worker_test_base.create_cluster_spec(num_workers=2)
122 cluster_spec = multi_worker_test_base.create_cluster_spec(num_workers=2)
149 cluster_spec = multi_worker_test_base.create_cluster_spec(num_workers=2)
170 cluster_spec = multi_worker_test_base.create_cluster_spec(num_workers=2)
209 cluster_spec = multi_worker_test_base.create_cluster_spec(num_workers=2)
/external/tensorflow/tensorflow/python/keras/distribute/
Dmulti_worker_callback_tf2_test.py144 cluster_spec=test_base.create_cluster_spec(num_workers=2),
170 cluster_spec=test_base.create_cluster_spec(num_workers=2),
237 cluster_spec=test_base.create_cluster_spec(num_workers=2),
275 cluster_spec=test_base.create_cluster_spec(num_workers=2),
304 cluster_spec=test_base.create_cluster_spec(num_workers=2),
333 cluster_spec=test_base.create_cluster_spec(num_workers=2),
362 cluster_spec=test_base.create_cluster_spec(num_workers=2),
Dmulti_worker_test.py198 cluster_spec = test_base.create_cluster_spec(num_workers=num_workers)
/external/tensorflow/tensorflow/python/kernel_tests/
Dcollective_ops_multi_worker_test.py88 cluster_spec = multi_worker_test_base.create_cluster_spec(num_workers=2)
100 cluster_spec = multi_worker_test_base.create_cluster_spec(num_workers=2)
138 cluster_spec = multi_worker_test_base.create_cluster_spec(
152 cluster_spec = multi_worker_test_base.create_cluster_spec(num_workers=2)
160 multi_worker_test_base.create_cluster_spec(num_workers=2),
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.__internal__.distribute.multi_process_runner.pbtxt16 name: "create_cluster_spec"
/external/tensorflow/tensorflow/python/keras/integration_test/
Dmulti_worker_tutorial_test.py214 tf.__internal__.distribute.multi_process_runner.create_cluster_spec(
339 tf.__internal__.distribute.multi_process_runner.create_cluster_spec(