Home
last modified time | relevance | path

Searched refs:worker_names (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/mpi/
Dmpi_utils.cc36 std::vector<char> worker_names(number_of_procs * max_worker_name_length); in MPIUtils() local
38 &worker_names[0], max_worker_name_length, MPI_CHAR, in MPIUtils()
43 name_to_id_[std::string(&worker_names[i * 128])] = i; in MPIUtils()
46 << "\tgRPC-name: " << std::string(&worker_names[i * 128]) in MPIUtils()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_session.cc1235 const std::vector<string> worker_names = filtered_worker_list_; in CreateWorkerSessions() local
1250 BlockingCounter done(worker_names.size()); in CreateWorkerSessions()
1251 std::vector<WorkerGroup> workers(worker_names.size()); in CreateWorkerSessions()
1264 for (size_t i = 0; i < worker_names.size(); ++i) { in CreateWorkerSessions()
1265 workers[i].name = &worker_names[i]; in CreateWorkerSessions()
1266 workers[i].worker = worker_cache->CreateWorker(worker_names[i]); in CreateWorkerSessions()
1270 if (!DeviceNameUtils::ParseFullName(worker_names[i], &name)) { in CreateWorkerSessions()
1271 status = errors::Internal("Could not parse name ", worker_names[i]); in CreateWorkerSessions()
1276 status = errors::Internal("Incomplete worker name ", worker_names[i]); in CreateWorkerSessions()
1298 for (size_t i = 0; i < worker_names.size(); ++i) { in CreateWorkerSessions()
[all …]
Dmaster.cc620 std::vector<string> worker_names; in CleanupWorkers() local
622 env_->worker_cache, &worker_names); in CleanupWorkers()
623 if (!worker_names.empty()) { in CleanupWorkers()
624 const int num_workers = worker_names.size(); in CleanupWorkers()
631 const string& worker_name = worker_names[i]; in CleanupWorkers()