Home
last modified time | relevance | path

Searched refs:server_def (Results 1 – 25 of 67) sorted by relevance

123

/external/tensorflow/tensorflow/core/distributed_runtime/
Dsession_mgr_test.cc61 [](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in __anon73e895030102() argument
69 ServerDef server_def; in TEST_F() local
70 server_def.set_job_name("worker"); in TEST_F()
71 server_def.set_task_index(3); in TEST_F()
74 TF_EXPECT_OK(mgr_.CreateSession(session_handle, server_def, true)); in TEST_F()
83 ServerDef server_def; in TEST_F() local
84 server_def.set_job_name("worker"); in TEST_F()
85 server_def.set_task_index(3); in TEST_F()
86 auto job = server_def.mutable_cluster()->add_job(); in TEST_F()
97 TF_EXPECT_OK(mgr_.CreateSession(session_handle, server_def, in TEST_F()
[all …]
Dserver_lib_test.cc25 bool AcceptsOptions(const ServerDef& server_def) override { in AcceptsOptions() argument
26 return server_def.protocol() == "test_protocol"; in AcceptsOptions()
29 Status NewServer(const ServerDef& server_def, const Options& options, in NewServer() argument
37 ServerDef server_def; in TEST() local
38 server_def.set_protocol("test_protocol"); in TEST()
40 TF_EXPECT_OK(NewServer(server_def, &server)); in TEST()
44 ServerDef server_def; in TEST() local
45 server_def.set_protocol("fake_protocol"); in TEST()
47 Status s = NewServer(server_def, &server); in TEST()
Dserver_lib.cc49 Status ServerFactory::GetFactory(const ServerDef& server_def, in GetFactory() argument
53 if (server_factory.second->AcceptsOptions(server_def)) { in GetFactory()
66 server_def.DebugString(), "\nThe available server factories are: [ ", in GetFactory()
72 Status NewServer(const ServerDef& server_def, in NewServer() argument
75 TF_RETURN_IF_ERROR(ServerFactory::GetFactory(server_def, &factory)); in NewServer()
76 return factory->NewServer(server_def, ServerFactory::Options(), out_server); in NewServer()
81 Status NewServerWithOptions(const ServerDef& server_def, in NewServerWithOptions() argument
85 TF_RETURN_IF_ERROR(ServerFactory::GetFactory(server_def, &factory)); in NewServerWithOptions()
86 return factory->NewServer(server_def, options, out_server); in NewServerWithOptions()
Dmaster_env.h50 WorkerCacheFactoryOptions(const ServerDef& server_def) { in WorkerCacheFactoryOptions()
51 if (server_def.has_cluster() && !server_def.job_name().empty()) { in WorkerCacheFactoryOptions()
52 cluster_def = &server_def.cluster(); in WorkerCacheFactoryOptions()
53 job_name = &server_def.job_name(); in WorkerCacheFactoryOptions()
54 task_index = server_def.task_index(); in WorkerCacheFactoryOptions()
55 protocol = &server_def.protocol(); in WorkerCacheFactoryOptions()
56 rpc_options = &server_def.default_session_config().rpc_options(); in WorkerCacheFactoryOptions()
Dsession_mgr.cc58 std::string SessionMgr::WorkerNameFromServerDef(const ServerDef& server_def) { in WorkerNameFromServerDef() argument
59 return strings::StrCat("/job:", server_def.job_name(), in WorkerNameFromServerDef()
60 "/replica:0/task:", server_def.task_index()); in WorkerNameFromServerDef()
64 const ServerDef& server_def, in CreateSession() argument
67 return CreateSession(session, server_def, {}, isolate_session_state, in CreateSession()
73 const std::string& session, const ServerDef& server_def, in CreateSession() argument
77 return CreateSession(session, server_def, cluster_device_attributes, in CreateSession()
84 const std::string& session, const ServerDef& server_def, in CreateSession() argument
121 if (server_def.cluster().job().empty()) { in CreateSession()
125 TF_RETURN_IF_ERROR(worker_cache_factory_(server_def, &worker_cache)); in CreateSession()
[all …]
Dserver_lib.h70 virtual Status UpdateServerDef(const ServerDef& server_def) = 0;
98 virtual Status NewServer(const ServerDef& server_def, const Options& options,
103 virtual bool AcceptsOptions(const ServerDef& server_def) = 0;
116 static Status GetFactory(const ServerDef& server_def,
122 Status NewServer(const ServerDef& server_def,
124 Status NewServerWithOptions(const ServerDef& server_def,
Dsession_mgr.h54 const std::string& session, const ServerDef& server_def,
60 const std::string& session, const ServerDef& server_def,
72 const std::string& session, const ServerDef& server_def,
84 Status UpdateSession(const std::string& session, const ServerDef& server_def,
100 static std::string WorkerNameFromServerDef(const ServerDef& server_def);
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_remote_test.cc29 tensorflow::ServerDef server_def; in GetServerDef() local
30 server_def.set_protocol("grpc"); in GetServerDef()
31 server_def.set_job_name(job_name); in GetServerDef()
32 server_def.set_task_index(0); in GetServerDef()
33 tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); in GetServerDef()
41 return server_def; in GetServerDef()
54 tensorflow::ServerDef server_def = GetServerDef("worker", 3); in TEST() local
57 std::string serialized = server_def.SerializeAsString(); in TEST()
59 server_def.set_task_index(1); in TEST()
62 server_def, tensorflow::Env::Default(), &worker_server1) in TEST()
[all …]
/external/tensorflow/tensorflow/c/eager/
Dc_api_cluster_test.cc33 void ReplaceTaskInServerDef(tensorflow::ServerDef* server_def, int task_index) { in ReplaceTaskInServerDef() argument
34 tensorflow::JobDef* job_def = server_def->mutable_cluster()->mutable_job(0); in ReplaceTaskInServerDef()
108 tensorflow::ServerDef server_def = GetServerDef(2); in TestRemoteExecuteChangeServerDef() local
111 string serialized = server_def.SerializeAsString(); in TestRemoteExecuteChangeServerDef()
113 server_def.set_task_index(1); in TestRemoteExecuteChangeServerDef()
117 server_def, tensorflow::Env::Default(), &worker_server) in TestRemoteExecuteChangeServerDef()
202 tensorflow::ServerDef server_def = GetServerDef(2); in TestRemoteExecuteUpdateServerDef() local
204 string serialized = server_def.SerializeAsString(); in TestRemoteExecuteUpdateServerDef()
206 server_def.set_task_index(1); in TestRemoteExecuteUpdateServerDef()
209 server_def, tensorflow::Env::Default(), &worker_server) in TestRemoteExecuteUpdateServerDef()
[all …]
Dc_api_distributed_test.cc125 tensorflow::ServerDef server_def = GetServerDef(3); in TestFunctionWithPackedInput() local
128 string serialized = server_def.SerializeAsString(); in TestFunctionWithPackedInput()
130 server_def.set_task_index(1); in TestFunctionWithPackedInput()
133 server_def, tensorflow::Env::Default(), &worker_server1) in TestFunctionWithPackedInput()
137 server_def.set_task_index(2); in TestFunctionWithPackedInput()
140 server_def, tensorflow::Env::Default(), &worker_server2) in TestFunctionWithPackedInput()
349 tensorflow::ServerDef server_def = GetServerDef(3); in TEST() local
351 string serialized = server_def.SerializeAsString(); in TEST()
353 server_def.set_task_index(1); in TEST()
356 server_def, tensorflow::Env::Default(), &worker_server1) in TEST()
[all …]
Dc_api_remote_test_util.cc74 tensorflow::ServerDef server_def = GetServerDef(3); in TestRemoteExecuteSilentCopies() local
77 string serialized = server_def.SerializeAsString(); in TestRemoteExecuteSilentCopies()
79 server_def.set_task_index(1); in TestRemoteExecuteSilentCopies()
82 server_def, tensorflow::Env::Default(), &worker_server1) in TestRemoteExecuteSilentCopies()
86 server_def.set_task_index(2); in TestRemoteExecuteSilentCopies()
89 server_def, tensorflow::Env::Default(), &worker_server2) in TestRemoteExecuteSilentCopies()
Dc_api_test_util.cc426 tensorflow::ServerDef server_def; in GetServerDef() local
427 server_def.set_protocol("grpc"); in GetServerDef()
428 server_def.set_job_name(job_name); in GetServerDef()
429 server_def.set_task_index(0); in GetServerDef()
430 tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); in GetServerDef()
438 return server_def; in GetServerDef()
448 tensorflow::ServerDef server_def; in GetMultiClientServerDef() local
449 server_def.set_protocol("grpc"); in GetMultiClientServerDef()
450 server_def.set_job_name(job_name); in GetMultiClientServerDef()
451 server_def.set_task_index(0); in GetMultiClientServerDef()
[all …]
Dc_api_remote_test.cc41 tensorflow::ServerDef server_def = GetServerDef(2); in TestRemoteExecute() local
44 string serialized = server_def.SerializeAsString(); in TestRemoteExecute()
46 server_def.set_task_index(1); in TestRemoteExecute()
50 server_def, tensorflow::Env::Default(), &worker_server) in TestRemoteExecute()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_server_lib.cc99 GrpcServer::GrpcServer(const ServerDef& server_def, Env* env) in GrpcServer() argument
100 : env_(env), state_(NEW), server_def_(server_def) {} in GrpcServer()
137 Status GrpcServer::GetHostAndPort(const ServerDef& server_def, in GetHostAndPort() argument
141 for (const auto& job : server_def.cluster().job()) { in GetHostAndPort()
142 if (job.name() == server_def.job_name()) { in GetHostAndPort()
143 auto iter = job.tasks().find(server_def.task_index()); in GetHostAndPort()
145 return errors::Internal("Task ", server_def.task_index(), in GetHostAndPort()
147 server_def.job_name(), "\""); in GetHostAndPort()
150 if (server_def.port() != 0) { in GetHostAndPort()
151 *port = server_def.port(); in GetHostAndPort()
[all …]
Dgrpc_server_lib.h77 GrpcServer(const ServerDef& server_def, Env* env);
78 GrpcServer(const ServerDef& server_def, DeviceMgr* local_device_mgr,
88 static Status Create(const ServerDef& server_def, Env* env,
90 static Status Create(const ServerDef& server_def, Env* env,
93 static Status Create(const ServerDef& server_def, Env* env,
116 Status UpdateServerDef(const ServerDef& server_def) override;
125 virtual Status GetHostAndPort(const ServerDef& server_def, string* host_name,
131 const ServerDef& server_def) const;
169 const ServerDef& server_def() const { return server_def_; } in server_def() function
/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
Dc_api_session_coordination_test.cc41 void EnableCoordinationService(tensorflow::ServerDef* server_def) { in EnableCoordinationService() argument
42 auto coord_config = server_def->mutable_default_session_config() in EnableCoordinationService()
66 tensorflow::ServerDef server_def = GetServerDef(job_name, num_workers); in TEST_P() local
71 strings::StrCat("grpc://", server_def.cluster().job(0).tasks().at(0)); in TEST_P()
73 EnableCoordinationService(&server_def); in TEST_P()
74 server_def.mutable_default_session_config() in TEST_P()
81 ASSERT_TRUE(ServerFactory::GetFactory(server_def, &factory).ok()); in TEST_P()
84 server_def.set_task_index(worker_id); in TEST_P()
86 ->NewServer(server_def, ServerFactory::Options(), in TEST_P()
140 ConfigProto configs = server_def.default_session_config(); in TEST_P()
[all …]
Dc_api_coordination_test.cc45 tensorflow::ServerDef* server_def, in ConfigCoordinationService() argument
47 auto coord_config = server_def->mutable_default_session_config() in ConfigCoordinationService()
115 tensorflow::ServerDef server_def = in TEST() local
119 ConfigCoordinationService(&server_def, in TEST()
122 tensorflow::ServerDef server_def_copy = server_def; in TEST()
184 tensorflow::ServerDef server_def = in TEST() local
186 ConfigCoordinationService(&server_def); in TEST()
189 tensorflow::ServerDef server_def_copy = server_def; in TEST()
271 tensorflow::ServerDef server_def = in TEST() local
273 ConfigCoordinationService(&server_def); in TEST()
[all …]
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt_distributed_impl.cc47 const tensorflow::ServerDef& server_def) { in ConvertServerDefToDistributedConfiguration() argument
49 dist_config.set_job_name(server_def.job_name()); in ConvertServerDefToDistributedConfiguration()
50 dist_config.set_task_id(server_def.task_index()); in ConvertServerDefToDistributedConfiguration()
56 server_def.cluster().job(0).name(), /*task_id=*/0)); in ConvertServerDefToDistributedConfiguration()
57 for (const auto& job_def : server_def.cluster().job()) { in ConvertServerDefToDistributedConfiguration()
90 const tensorflow::ServerDef& server_def, bool reset_context,
94 const tensorflow::ServerDef& server_def) override;
118 const tensorflow::ServerDef& server_def, bool reset_context, in SetOrUpdateServerDef() argument
122 ConvertServerDefToDistributedConfiguration(server_def); in SetOrUpdateServerDef()
160 const tensorflow::ServerDef& server_def) { in EnableCollectiveOps() argument
[all …]
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext_distributed_manager.cc65 const ServerDef& server_def) { in AreLocalDevicesCompatible() argument
66 if (server_def.job_name() != context->HostCPU()->parsed_name().job) { in AreLocalDevicesCompatible()
69 return server_def.default_session_config().SerializeAsString() == in AreLocalDevicesCompatible()
177 uint64 context_view_id, const ServerDef& server_def, in GetReplacedFromExistingWorkers() argument
216 int keep_alive_secs, const ServerDef& server_def, in CreateRemoteContexts() argument
248 *request.mutable_server_def() = server_def; in CreateRemoteContexts()
252 server_def.default_session_config()); in CreateRemoteContexts()
295 const ServerDef& server_def, in UpdateRemoteContexts() argument
365 *request.mutable_server_def() = server_def; in UpdateRemoteContexts()
369 server_def.default_session_config()); in UpdateRemoteContexts()
[all …]
/external/tensorflow/tensorflow/python/eager/
Dremote_cluster_test.py54 server_def = tensorflow_server_pb2.ServerDef(
59 server_def.default_session_config.experimental.coordination_config.service_type = "standalone"
61 return server_def
132 server_def=get_server_def(
168 context.update_server_def(server_def=self.server_def_s1_s2_s3)
194 context.update_server_def(server_def=self.server_def_s1)
214 context.update_server_def(server_def=self.server_def_s1_s3)
232 context.update_server_def(server_def=self.server_def_s1_s2_s3)
257 context.update_server_def(server_def=self.server_def_s1)
285 context.update_server_def(server_def=self.server_def_s1_s2_s3)
[all …]
/external/tensorflow/tensorflow/dtensor/python/
Dmulti_client_util.py96 server_def = tensorflow_server_pb2.ServerDef(
103 server_def.default_session_config.rpc_options.num_channels_per_target = 4
104 server_def.default_session_config.experimental.recv_buf_max_chunk = -1
106 logging.info("Enabling collectives with server_def: %s", server_def)
107 context.context().enable_collective_ops(server_def)
/external/tensorflow/tensorflow/python/training/
Dserver_lib.py53 server_def = tensorflow_server_pb2.ServerDef()
55 server_def.MergeFrom(server_or_cluster_def)
57 server_def.job_name = job_name
59 server_def.task_index = task_index
61 server_def.protocol = protocol
63 server_def.default_session_config.MergeFrom(config)
84 server_def = tensorflow_server_pb2.ServerDef(
90 server_def.default_session_config.MergeFrom(config)
91 return server_def
188 def server_def(self): member in Server
/external/tensorflow/tensorflow/python/debug/lib/
Dgrpc_tensorflow_server.py90 server_def = tensorflow_server_pb2.ServerDef(protocol="grpc")
93 parse_cluster_spec(FLAGS.cluster_spec, server_def.cluster, FLAGS.verbose)
98 server_def.job_name = FLAGS.job_name
103 server_def.task_index = FLAGS.task_id
109 server = server_lib.Server(server_def, config=config)
/external/tensorflow/tensorflow/core/distributed_runtime/preemption/
Dpreemption_sync_manager_test.cc141 ServerDef server_def; in EnableCoordinationService() local
142 server_def.set_protocol("grpc"); in EnableCoordinationService()
143 server_def.set_job_name(kJobName); in EnableCoordinationService()
144 server_def.set_task_index(0); in EnableCoordinationService()
145 auto job_def = server_def.mutable_cluster()->add_job(); in EnableCoordinationService()
149 auto coordination_config = server_def.mutable_default_session_config() in EnableCoordinationService()
154 "standalone", Env::Default(), server_def, /*cache=*/nullptr); in EnableCoordinationService()
/external/tensorflow/tensorflow/java/src/main/native/
Dserver_jni.cc38 JNIEnv* env, jclass clazz, jbyteArray server_def) { in Java_org_tensorflow_Server_allocate() argument
41 jbyte* server_def_ptr = env->GetByteArrayElements(server_def, nullptr); in Java_org_tensorflow_Server_allocate()
44 server_def_ptr, static_cast<size_t>(env->GetArrayLength(server_def)), in Java_org_tensorflow_Server_allocate()
47 env->ReleaseByteArrayElements(server_def, server_def_ptr, JNI_ABORT); in Java_org_tensorflow_Server_allocate()

123