Searched refs:WorkerGrpcDataServer (Results 1 – 6 of 6) sorted by relevance
48 py::class_<tensorflow::data::WorkerGrpcDataServer>(m, "WorkerGrpcDataServer") in PYBIND11_MODULE()49 .def("start", &tensorflow::data::WorkerGrpcDataServer::Start) in PYBIND11_MODULE()50 .def("stop", &tensorflow::data::WorkerGrpcDataServer::Stop) in PYBIND11_MODULE()51 .def("join", &tensorflow::data::WorkerGrpcDataServer::Join, in PYBIND11_MODULE()53 .def("bound_port", &tensorflow::data::WorkerGrpcDataServer::BoundPort) in PYBIND11_MODULE()55 [](tensorflow::data::WorkerGrpcDataServer* server) -> int { in PYBIND11_MODULE()82 -> std::unique_ptr<tensorflow::data::WorkerGrpcDataServer> { in PYBIND11_MODULE()88 std::unique_ptr<tensorflow::data::WorkerGrpcDataServer> server; in PYBIND11_MODULE()
117 WorkerGrpcDataServer::WorkerGrpcDataServer( in WorkerGrpcDataServer() function in tensorflow::data::WorkerGrpcDataServer122 WorkerGrpcDataServer::~WorkerGrpcDataServer() { delete service_; } in ~WorkerGrpcDataServer()124 void WorkerGrpcDataServer::AddDataServiceToBuilder( in AddDataServiceToBuilder()129 Status WorkerGrpcDataServer::StartServiceInternal() { in StartServiceInternal()154 Status WorkerGrpcDataServer::NumTasks(int* num_tasks) { in NumTasks()173 std::unique_ptr<WorkerGrpcDataServer>& out_server) { in NewWorkerServer()174 out_server = absl::make_unique<WorkerGrpcDataServer>(config); in NewWorkerServer()
97 class WorkerGrpcDataServer : public GrpcDataServerBase {99 explicit WorkerGrpcDataServer(const experimental::WorkerConfig& config);100 ~WorkerGrpcDataServer() override;122 std::unique_ptr<WorkerGrpcDataServer>& out_server);
47 std::vector<std::unique_ptr<WorkerGrpcDataServer>> workers_;
64 std::unique_ptr<WorkerGrpcDataServer> worker; in AddWorker()
97 tensorflow::data::WorkerGrpcDataServer::NumTasks