Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dsession_mgr_test.cc86 auto job = server_def.mutable_cluster()->add_job(); in TEST_F()
162 auto job = server_def.mutable_cluster()->add_job(); in TEST_F()
215 auto job = server_def.mutable_cluster()->add_job(); in TEST_F()
H A Drpcbench_test.cc75 auto job_def = server.mutable_cluster()->add_job(); in MakeGRPCCluster()
H A Dmaster_session.cc1368 *workers[i].request.mutable_server_def()->mutable_cluster() = cluster_def; in CreateWorkerSessions()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_cluster_test.cc34 tensorflow::JobDef* job_def = server_def->mutable_cluster()->mutable_job(0); in ReplaceTaskInServerDef()
379 tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); in TestRemoteExecuteUpdateServerDefWithFailures()
451 server_def.mutable_cluster()->mutable_job(0)->set_name(second_name); in TestConnectToCluster()
452 (*server_def.mutable_cluster()->mutable_job(0)->mutable_tasks())[0] = in TestConnectToCluster()
H A Dc_api_test_util.cc430 tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); in GetServerDef()
452 tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); in GetMultiClientServerDef()
H A Dc_api_test.cc1973 tensorflow::ClusterDef* cluster_def = server_def_copy.mutable_cluster(); in ReplaceTaskInServerDef()
2185 tensorflow::JobDef* job_def = server_def->mutable_cluster()->mutable_job(0); in ReplaceTaskInServerDef()
2338 single_host_server_def.mutable_cluster(); in CreateSingleHostServerDef()
2363 tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); in GetClusterServerDef()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Dgrpc_testlib_server.cc47 JobDef* job_def = options->mutable_cluster()->add_job(); in FillServerDef()
H A Dgrpc_tensorflow_server.cc49 ClusterDef* const cluster = options->mutable_cluster(); in FillServerDef()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Dtensorflow_server.pb.h211 ::tensorflow::ClusterDef* mutable_cluster(); in Swap()
360 inline ::tensorflow::ClusterDef* ServerDef::mutable_cluster() { in mutable_cluster() function
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
H A Dc_api_session_coordination_test.cc144 server_def.mutable_cluster()->mutable_job(0)->set_name("worker"); in TEST_P()
H A Dc_api_coordination_test.cc448 auto cluster = server_def.mutable_cluster(); in TEST_P()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/
H A Dparallel_device_remote_test.cc33 tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); in GetServerDef()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/coordination/
H A Dcoordination_service_test.cc301 ClusterDef* cluster_def = server_def.mutable_cluster(); in TEST()
364 JobDef* job_def = server_def.mutable_cluster()->add_job(); in TEST()
385 JobDef* job_def = server_def.mutable_cluster()->add_job(); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/preemption/
H A Dpreemption_sync_manager_test.cc145 auto job_def = server_def.mutable_cluster()->add_job(); in EnableCoordinationService()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
H A Dservice.cc53 auto job_def = server_def.mutable_cluster()->add_job(); in EnableCoordinationService()