Searched refs:GrpcServer (Results 1 – 12 of 12) sorted by relevance
94 GrpcServer::GrpcServer(const ServerDef& server_def, Env* env) in GrpcServer() function in tensorflow::GrpcServer97 GrpcServer::~GrpcServer() { in ~GrpcServer()126 void GrpcServer::MaybeMutateBuilder(::grpc::ServerBuilder* builder) {} in MaybeMutateBuilder()129 Status GrpcServer::GetHostAndPort(const ServerDef& server_def, in GetHostAndPort()169 Status GrpcServer::Init(const GrpcServerOptions& opts) { in Init()336 Status GrpcServer::ParseChannelSpec(const WorkerCacheFactoryOptions& options, in ParseChannelSpec()359 Status GrpcServer::WorkerCacheFactory(const WorkerCacheFactoryOptions& options, in WorkerCacheFactory()397 Status GrpcServer::Start() { in Start()424 Status GrpcServer::AddMasterEagerContextToEagerService( in AddMasterEagerContextToEagerService()431 Status GrpcServer::UpdateServerDef(const ServerDef& server_def) { in UpdateServerDef()[all …]
75 class GrpcServer : public ServerInterface {77 GrpcServer(const ServerDef& server_def, Env* env);78 GrpcServer(const ServerDef& server_def, DeviceMgr* local_device_mgr,88 std::unique_ptr<GrpcServer>* out_server);96 virtual ~GrpcServer();
60 std::unique_ptr<tensorflow::GrpcServer> worker_server1; in TEST()61 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TEST()67 std::unique_ptr<tensorflow::GrpcServer> worker_server2; in TEST()68 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TEST()98 std::unique_ptr<tensorflow::GrpcServer> worker_server1; in TEST()99 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TEST()105 std::unique_ptr<tensorflow::GrpcServer> worker_server2; in TEST()106 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TEST()
131 std::unique_ptr<tensorflow::GrpcServer> worker_server1; in TestFunctionWithPackedInput()132 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TestFunctionWithPackedInput()138 std::unique_ptr<tensorflow::GrpcServer> worker_server2; in TestFunctionWithPackedInput()139 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TestFunctionWithPackedInput()354 std::unique_ptr<tensorflow::GrpcServer> worker_server1; in TEST()355 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TEST()360 std::unique_ptr<tensorflow::GrpcServer> worker_server2; in TEST()361 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TEST()461 std::unique_ptr<tensorflow::GrpcServer> worker_server1; in TestDistributedFunctionCancellation()462 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TestDistributedFunctionCancellation()[all …]
115 std::unique_ptr<tensorflow::GrpcServer> worker_server; in TestRemoteExecuteChangeServerDef()116 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TestRemoteExecuteChangeServerDef()151 tensorflow::Status s = tensorflow::GrpcServer::Create( in TestRemoteExecuteChangeServerDef()207 std::unique_ptr<tensorflow::GrpcServer> worker_server; in TestRemoteExecuteUpdateServerDef()208 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TestRemoteExecuteUpdateServerDef()255 std::unique_ptr<tensorflow::GrpcServer> worker_server; in TestRemoteExecuteUpdateServerDefResourceAccess()256 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TestRemoteExecuteUpdateServerDefResourceAccess()289 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TestRemoteExecuteUpdateServerDefResourceAccess()355 std::unique_ptr<tensorflow::GrpcServer> worker_server; in TestRemoteExecuteUpdateServerDefWithFailures()356 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TestRemoteExecuteUpdateServerDefWithFailures()
80 std::unique_ptr<tensorflow::GrpcServer> worker_server1; in TestRemoteExecuteSilentCopies()81 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TestRemoteExecuteSilentCopies()87 std::unique_ptr<tensorflow::GrpcServer> worker_server2; in TestRemoteExecuteSilentCopies()88 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TestRemoteExecuteSilentCopies()
48 std::unique_ptr<tensorflow::GrpcServer> worker_server; in TestRemoteExecute()49 ASSERT_TRUE(tensorflow::GrpcServer::Create( in TestRemoteExecute()
88 Dagger.GrpcServer.SERVICE_DEFINITION_FACTORY, "serviceDefinitionFactory") in provideServiceDefinitionContribution()98 Dagger.GrpcServer.PROXY_SERVER_CALL_HANDLER, in provideServiceDefinitionContribution()154 .returns(Dagger.GrpcServer.SERVICE_DEFINITION_FACTORY) in provideServiceDefinitionFactory()166 .addSuperinterface(Dagger.GrpcServer.SERVICE_DEFINITION_FACTORY) in anonymousServiceDefinitionFactory()175 Dagger.GrpcServer.GRPC_CALL_METADATA_MODULE, in anonymousServiceDefinitionFactory()
70 protected static final class GrpcServer { class in SourceGenerator.Dagger71 private GrpcServer() {} in GrpcServer() method in SourceGenerator.Dagger.GrpcServer
56 Dagger.GrpcServer.GRPC_CALL_METADATA_MODULE, "grpcCallMetadataModule") in createType()
90 .addParameter(Dagger.GrpcServer.GRPC_CALL_METADATA_MODULE, "grpcCallMetadataModule") in unscopedComponentFactory()
129 tensorflow::GrpcServer* grpc_server = in ListRemoteWorkers()130 dynamic_cast<tensorflow::GrpcServer*>(server); in ListRemoteWorkers()424 tensorflow::GrpcServer* grpc_server; in UpdateContextWithServerDef()432 grpc_server = dynamic_cast<tensorflow::GrpcServer*>(new_server.get()); in UpdateContextWithServerDef()440 grpc_server = dynamic_cast<tensorflow::GrpcServer*>(context->GetServer()); in UpdateContextWithServerDef()695 tensorflow::GrpcServer* grpc_server = in EnableCollectiveOps()696 dynamic_cast<tensorflow::GrpcServer*>(context_->GetServer()); in EnableCollectiveOps()700 grpc_server = dynamic_cast<tensorflow::GrpcServer*>(new_server.get()); in EnableCollectiveOps()723 GrpcServer* grpc_server = dynamic_cast<GrpcServer*>(context_->GetServer()); in CheckRemoteAlive()