Home
last modified time | relevance | path

Searched refs:create_local_cluster (Results 1 – 20 of 20) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Diterator_cluster_test.py45 worker, _ = test_util.create_local_cluster(
101 worker, _ = test_util.create_local_cluster(
110 workers, _ = test_util.create_local_cluster(2, 1)
118 worker, _ = test_util.create_local_cluster(1, 1)
156 worker, _ = test_util.create_local_cluster(1, 1)
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dlocalhost_cluster_performance_test.py36 workers, _ = test.create_local_cluster(num_workers=2, num_ps=2)
63 test.create_local_cluster(num_workers=1, num_ps=10)
79 workers, _ = test.create_local_cluster(num_workers=1, num_ps=100)
H A Dsync_replicas_optimizer_test.py23 from tensorflow.python.framework.test_util import create_local_cluster
95 workers, _ = create_local_cluster(num_workers=num_workers, num_ps=num_ps)
187 workers, _ = create_local_cluster(num_workers=num_workers, num_ps=num_ps)
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dremote_test.py58 workers, _ = test_util.create_local_cluster(1, 0)
217 workers, _ = test_util.create_local_cluster(1, 0)
238 workers, _ = test_util.create_local_cluster(1, 0)
326 workers, _ = test_util.create_local_cluster(3, 0)
609 workers, ps = test_util.create_local_cluster(num_workers=2, num_ps=2)
H A Ddevice_placement_test.py223 workers, _ = test_util.create_local_cluster(2, 0)
/aosp_15_r20/external/tensorflow/tensorflow/python/platform/
H A Dtest.py25 from tensorflow.python.framework.test_util import create_local_cluster
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.test.pbtxt24 name: "create_local_cluster"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.test.pbtxt36 name: "create_local_cluster"
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dstrategy_test_lib.py600 workers, _ = test_util.create_local_cluster(2, 0)
613 workers, _ = test_util.create_local_cluster(2, 0)
H A Dmulti_worker_test_base.py563 workers, _ = test_util.create_local_cluster(num_workers=1, num_ps=0)
H A Ddistribute_coordinator_test.py160 cls._workers, cls._ps = test_util.create_local_cluster(
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
H A DBUILD263 flaky = 1, # create_local_cluster sometimes times out.
H A Dlist_ops_test.py874 worker = test_util.create_local_cluster(num_workers=1, num_ps=1)[0][0]
888 worker = test_util.create_local_cluster(num_workers=1, num_ps=1)[0][0]
903 worker = test_util.create_local_cluster(num_workers=1, num_ps=1)[0][0]
917 worker = test_util.create_local_cluster(num_workers=1, num_ps=1)[0][0]
H A Dtensor_array_ops_test.py73 cls._workers, _ = test.create_local_cluster(num_workers=3, num_ps=0)
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dfunctional_ops_test.py412 worker, _ = test_util.create_local_cluster(
551 workers, _ = test_util.create_local_cluster(2, 1)
1045 workers, _ = test_util.create_local_cluster(2, 0)
H A Dpy_func_test.py555 workers, _ = test_util.create_local_cluster(num_workers=1, num_ps=0)
H A Dcond_v2_test.py1480 workers, _ = test_util.create_local_cluster(num_workers=1, num_ps=0)
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
H A Dreplicate_test.py339 worker, _ = test_util.create_local_cluster(
/aosp_15_r20/external/tensorflow/tensorflow/python/profiler/
H A Dmodel_analyzer_test.py677 workers, _ = test_util.create_local_cluster(2, 0)
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dtest_util.py3711 def create_local_cluster(num_workers, function