Searched refs:GetMultiClientServerDef (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/coordination/ |
| H A D | coordination_service_test.cc | 147 const ServerDef& server_def = GetMultiClientServerDef("worker", num_tasks); in CoordinationBarrierTest()
200 ServerDef server_def = GetMultiClientServerDef("worker", /*num_tasks=*/2); in EnableCoordinationService()
286 ServerDef server_def = GetMultiClientServerDef("chief", 1); in TEST()
363 ServerDef server_def = GetMultiClientServerDef("worker", 1); in TEST()
384 ServerDef server_def = GetMultiClientServerDef("worker", 1); in TEST()
518 const ServerDef& server_def = GetMultiClientServerDef("worker", 1); in TEST()
621 const ServerDef& server_def = GetMultiClientServerDef("worker", 3); in TEST()
679 const ServerDef& server_def = GetMultiClientServerDef("worker", 3); in TEST()
746 const ServerDef& server_def = GetMultiClientServerDef("worker", 2); in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/ |
| H A D | c_api_coordination_test.cc | 116 GetMultiClientServerDef("worker", cluster_size); in TEST()
185 GetMultiClientServerDef("worker", cluster_size); in TEST()
272 GetMultiClientServerDef("worker", cluster_size); in TEST()
342 GetMultiClientServerDef("worker", cluster_size); in TEST()
|
| H A D | c_api_multi_client_test.cc | 38 GetMultiClientServerDef("worker", cluster_size, /*num_virtual_gpus=*/2); in StartWorkers()
|
| H A D | c_api_multi_client_function_test.cc | 193 GetMultiClientServerDef("worker", cluster_size); in TEST_P()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | c_api_test_util.h | 144 tensorflow::ServerDef GetMultiClientServerDef(const std::string& job_name,
|
| H A D | c_api_test_util.cc | 445 tensorflow::ServerDef GetMultiClientServerDef(const std::string& job_name, in GetMultiClientServerDef() function
|