Searched refs:DispatchGrpcDataServer (Results 1 – 5 of 5) sorted by relevance
33 py::class_<tensorflow::data::DispatchGrpcDataServer>(m, in PYBIND11_MODULE()35 .def("start", &tensorflow::data::DispatchGrpcDataServer::Start) in PYBIND11_MODULE()36 .def("stop", &tensorflow::data::DispatchGrpcDataServer::Stop) in PYBIND11_MODULE()37 .def("join", &tensorflow::data::DispatchGrpcDataServer::Join, in PYBIND11_MODULE()39 .def("bound_port", &tensorflow::data::DispatchGrpcDataServer::BoundPort) in PYBIND11_MODULE()41 [](tensorflow::data::DispatchGrpcDataServer* server) -> int { in PYBIND11_MODULE()65 -> std::unique_ptr<tensorflow::data::DispatchGrpcDataServer> { in PYBIND11_MODULE()71 std::unique_ptr<tensorflow::data::DispatchGrpcDataServer> server; in PYBIND11_MODULE()
89 DispatchGrpcDataServer::DispatchGrpcDataServer( in DispatchGrpcDataServer() function in tensorflow::data::DispatchGrpcDataServer94 DispatchGrpcDataServer::~DispatchGrpcDataServer() { delete service_; } in ~DispatchGrpcDataServer()96 void DispatchGrpcDataServer::AddDataServiceToBuilder( in AddDataServiceToBuilder()101 Status DispatchGrpcDataServer::StartServiceInternal() { in StartServiceInternal()105 Status DispatchGrpcDataServer::NumWorkers(int* num_workers) { in NumWorkers()167 std::unique_ptr<DispatchGrpcDataServer>& out_server) { in NewDispatchServer()168 out_server = absl::make_unique<DispatchGrpcDataServer>(config); in NewDispatchServer()
79 class DispatchGrpcDataServer : public GrpcDataServerBase {81 explicit DispatchGrpcDataServer(const experimental::DispatcherConfig& config);82 ~DispatchGrpcDataServer() override;118 std::unique_ptr<DispatchGrpcDataServer>& out_server);
45 std::unique_ptr<DispatchGrpcDataServer> dispatcher_;
96 tensorflow::data::DispatchGrpcDataServer::NumWorkers