Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/distributed_runtime/
Dsession_mgr_test.cc58 [](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in __anonc5de579e0102() argument
66 ServerDef server_def; in TEST_F() local
67 server_def.set_job_name("worker"); in TEST_F()
68 server_def.set_task_index(3); in TEST_F()
71 TF_EXPECT_OK(mgr_.CreateSession(session_handle, server_def, true)); in TEST_F()
80 ServerDef server_def; in TEST_F() local
81 server_def.set_job_name("worker"); in TEST_F()
82 server_def.set_task_index(3); in TEST_F()
83 auto job = server_def.mutable_cluster()->add_job(); in TEST_F()
88 TF_EXPECT_OK(mgr_.CreateSession(session_handle, server_def, true)); 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, 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()
Dmaster_env.h48 WorkerCacheFactoryOptions(const ServerDef& server_def) { in WorkerCacheFactoryOptions()
49 if (server_def.has_cluster() && !server_def.job_name().empty()) { in WorkerCacheFactoryOptions()
50 cluster_def = &server_def.cluster(); in WorkerCacheFactoryOptions()
51 job_name = &server_def.job_name(); in WorkerCacheFactoryOptions()
52 task_index = server_def.task_index(); in WorkerCacheFactoryOptions()
53 protocol = &server_def.protocol(); in WorkerCacheFactoryOptions()
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, out_server); in NewServer()
Dserver_lib.h69 virtual Status NewServer(const ServerDef& server_def,
74 virtual bool AcceptsOptions(const ServerDef& server_def) = 0;
87 static Status GetFactory(const ServerDef& server_def,
93 Status NewServer(const ServerDef& server_def,
Dsession_mgr.cc47 string SessionMgr::WorkerNameFromServerDef(const ServerDef& server_def) { in WorkerNameFromServerDef() argument
48 return strings::StrCat("/job:", server_def.job_name(), "/replica:0/task:", in WorkerNameFromServerDef()
49 server_def.task_index()); in WorkerNameFromServerDef()
53 const ServerDef& server_def, in CreateSession() argument
62 if (server_def.cluster().job().empty()) { in CreateSession()
66 TF_RETURN_IF_ERROR(worker_cache_factory_(server_def, &worker_cache)); in CreateSession()
67 worker_name = WorkerNameFromServerDef(server_def); in CreateSession()
Dsession_mgr.h49 Status CreateSession(const string& session, const ServerDef& server_def,
59 static string WorkerNameFromServerDef(const ServerDef& server_def);
/external/tensorflow/tensorflow/contrib/verbs/
Dverbs_server_lib.cc40 VerbsServer::VerbsServer(const ServerDef& server_def, Env* env) in VerbsServer() argument
41 : GrpcServer(server_def, env), verbs_state_(DISCONNECTED) {} in VerbsServer()
51 Status VerbsServer::ChannelCacheFactory(const ServerDef& server_def, in ChannelCacheFactory() argument
54 strings::StrCat("/job:", server_def.job_name(), "/replica:0", in ChannelCacheFactory()
55 "/task:", server_def.task_index()); in ChannelCacheFactory()
58 TF_RETURN_IF_ERROR(ParseChannelSpec(server_def, &channel_spec)); in ChannelCacheFactory()
91 CHECK(ChannelCacheFactory(server_def(), &channel_cache_).ok()); in Init()
133 Status VerbsServer::Create(const ServerDef& server_def, Env* env, in Create() argument
135 std::unique_ptr<VerbsServer> ret(new VerbsServer(server_def, Env::Default())); in Create()
149 bool AcceptsOptions(const ServerDef& server_def) override { in AcceptsOptions() argument
[all …]
Dverbs_server_lib.h29 VerbsServer(const ServerDef& server_def, Env* env);
32 static Status Create(const ServerDef& server_def, Env* env,
46 Status ChannelCacheFactory(const ServerDef& server_def,
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_server_lib.cc29 GdrServer::GdrServer(const ServerDef& server_def, Env* env) in GdrServer() argument
30 : GrpcServer(server_def, env) { in GdrServer()
33 for (const auto& job : server_def.cluster().job()) { in GdrServer()
34 if (job.name() == server_def.job_name()) { in GdrServer()
35 auto iter = job.tasks().find(server_def.task_index()); in GdrServer()
116 Status GdrServer::Create(const ServerDef& server_def, Env* env, in Create() argument
119 new GdrServer(server_def, env == nullptr ? Env::Default() : env)); in Create()
129 bool AcceptsOptions(const ServerDef& server_def) override { in AcceptsOptions() argument
130 return server_def.protocol() == "grpc+gdr"; in AcceptsOptions()
133 Status NewServer(const ServerDef& server_def, in NewServer() argument
[all …]
Dgdr_server_lib.h26 GdrServer(const ServerDef& server_def, Env* env);
29 static Status Create(const ServerDef& server_def, Env* env,
/external/tensorflow/tensorflow/contrib/mpi/
Dmpi_server_lib.cc47 MPIServer::MPIServer(const ServerDef& server_def, Env* env) in MPIServer() argument
48 : GrpcServer(server_def, env) {} in MPIServer()
75 Status MPIServer::Create(const ServerDef& server_def, Env* env, in Create() argument
77 std::unique_ptr<MPIServer> ret(new MPIServer(server_def, Env::Default())); in Create()
88 bool AcceptsOptions(const ServerDef& server_def) override { in AcceptsOptions() argument
89 return server_def.protocol() == "grpc+mpi"; in AcceptsOptions()
92 Status NewServer(const ServerDef& server_def, in NewServer() argument
94 return MPIServer::Create(server_def, Env::Default(), out_server); in NewServer()
Dmpi_server_lib.h30 MPIServer(const ServerDef& server_def, Env* env);
33 static Status Create(const ServerDef& server_def, Env* env,
47 Status ChannelCacheFactory(const ServerDef& server_def,
/external/tensorflow/tensorflow/python/training/
Dserver_lib.py58 server_def = tensorflow_server_pb2.ServerDef()
60 server_def.MergeFrom(server_or_cluster_def)
62 server_def.job_name = job_name
64 server_def.task_index = task_index
66 server_def.protocol = protocol
68 server_def.default_session_config.MergeFrom(config)
89 server_def = tensorflow_server_pb2.ServerDef(
93 server_def.default_session_config.MergeFrom(config)
94 return server_def
187 def server_def(self): member in Server
Dserver_lib_test.py190 self.assertEqual(0.1, server.server_def.default_session_config.gpu_options.
198 server_def = tensorflow_server_pb2.ServerDef(
203 server = server_lib.Server(server_def, config=config, start=False)
204 self.assertEqual(0.1, server.server_def.default_session_config.gpu_options.
347 server_def = tensorflow_server_pb2.ServerDef(
355 """, server_def)
365 server_def = tensorflow_server_pb2.ServerDef(
374 """, server_def)
385 server_def = tensorflow_server_pb2.ServerDef(
397 """, server_def)
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_server_lib.cc78 GrpcServer::GrpcServer(const ServerDef& server_def, Env* env) in GrpcServer() argument
79 : server_def_(server_def), env_(env), state_(NEW) {} in GrpcServer()
245 [this](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in Init() argument
246 WorkerCacheFactoryOptions options(server_def); in Init()
412 const ServerDef& server_def) const { in GetServerCredentials()
427 Status GrpcServer::Create(const ServerDef& server_def, Env* env, in Create() argument
430 new GrpcServer(server_def, env == nullptr ? Env::Default() : env)); in Create()
444 Status GrpcServer::Create(const ServerDef& server_def, Env* env, in Create() argument
447 new GrpcServer(server_def, env == nullptr ? Env::Default() : env)); in Create()
463 bool AcceptsOptions(const ServerDef& server_def) override { in AcceptsOptions() argument
[all …]
Dgrpc_server_lib.h74 GrpcServer(const ServerDef& server_def, Env* env);
80 static Status Create(const ServerDef& server_def, Env* env,
82 static Status Create(const ServerDef& server_def, Env* env,
105 const ServerDef& server_def) const;
123 const ServerDef& server_def() const { return server_def_; } in server_def() function
Dgrpc_tensorflow_server.cc115 tensorflow::ServerDef server_def; in main() local
117 task_index, &server_def); in main()
124 TF_QCHECK_OK(tensorflow::NewServer(server_def, &server)); in main()
/external/tensorflow/tensorflow/python/debug/lib/
Dgrpc_tensorflow_server.py93 server_def = tensorflow_server_pb2.ServerDef(protocol="grpc")
96 parse_cluster_spec(FLAGS.cluster_spec, server_def.cluster, FLAGS.verbose)
101 server_def.job_name = FLAGS.job_name
106 server_def.task_index = FLAGS.task_id
112 server = server_lib.Server(server_def, config=config)
/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()
/external/tensorflow/tensorflow/python/eager/
Dcontext.py226 server_def=None): argument
287 self._server_def = server_def
368 def set_server_def(self, server_def, keep_alive_secs=600): argument
387 if not server_def:
390 self._server_def = server_def
392 server_def_str = server_def.SerializeToString()
401 def enable_collective_ops(self, server_def): argument
413 if not server_def:
416 self._collective_ops_server_def = server_def
418 server_def_str = server_def.SerializeToString()
[all …]
Dremote.py72 server_def = ServerDef(
80 context.set_server_def(server_def)
/external/tensorflow/tensorflow/contrib/eager/python/
Dremote_test.py71 server_def = tensorflow_server_pb2.ServerDef(
77 return server_def
95 server_def=get_server_def(
170 server_def=get_server_def(
185 server_def=get_server_def(
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc115 int keep_alive_secs, const tensorflow::ServerDef& server_def, in CreateRemoteContexts() argument
130 *request.mutable_server_def() = server_def; in CreateRemoteContexts()
157 int keep_alive_secs, const tensorflow::ServerDef& server_def, in UpdateTFE_ContextWithServerDef() argument
173 tensorflow::strings::StrCat("/job:", server_def.job_name(), in UpdateTFE_ContextWithServerDef()
174 "/replica:0/task:", server_def.task_index()); in UpdateTFE_ContextWithServerDef()
177 LOG_AND_RETURN_IF_ERROR(tensorflow::NewServer(server_def, &server)); in UpdateTFE_ContextWithServerDef()
209 remote_workers, rendezvous_id, keep_alive_secs, server_def, in UpdateTFE_ContextWithServerDef()
217 session_name, server_def, true)); in UpdateTFE_ContextWithServerDef()
386 tensorflow::ServerDef server_def; in TFE_ContextSetServerDef() local
387 if (!server_def.ParseFromArray(proto, proto_len)) { in TFE_ContextSetServerDef()
[all …]
Dc_api_test.cc114 tensorflow::ServerDef server_def; in GetServerDef() local
115 server_def.set_protocol("grpc"); in GetServerDef()
116 server_def.set_job_name(job_name); in GetServerDef()
117 server_def.set_task_index(0); in GetServerDef()
118 tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); in GetServerDef()
126 return server_def; in GetServerDef()
134 tensorflow::ServerDef server_def = GetServerDef(2); in TestRemoteExecute() local
137 string serialized = server_def.SerializeAsString(); in TestRemoteExecute()
139 server_def.set_task_index(1); in TestRemoteExecute()
143 server_def, tensorflow::Env::Default(), &worker_server) in TestRemoteExecute()
[all …]

12