Searched refs:wc_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cancellable_call.h | 34 wc_(wc), in CancellableCall()
35 wi_(wc_->CreateWorker(remote_worker_)) {} in CancellableCall()
37 virtual ~CancellableCall() { wc_->ReleaseWorker(remote_worker_, wi_); } in ~CancellableCall()
59 WorkerCacheInterface* const wc_; // Not owned variable
|
| D | device_resolver_distributed_test.cc | 162 new TestableDeviceResolverDistributed(dev_mgr, &wc_, worker_name); in DefineWorker()
171 wc_.AddWorker(worker_name, fw); in DefineWorker()
174 FakeCache wc_; member in tensorflow::__anon5f4e7c180111::DeviceResDistTest
|
| D | collective_rma_distributed_test.cc | 199 device_mgrs_[0], dev_resolvers_[dev0_worker_name], &wc_, kStepId)); in SetUp()
227 new DeviceResolverDistributed(dev_mgr, &wc_, worker_name); in DefineWorker()
231 wc_.AddWorker(worker_name, fw); in DefineWorker()
242 FakeCache wc_; member in tensorflow::__anond64f02c00111::CollRMADistTest
|
| D | collective_param_resolver_distributed_test.cc | 172 new DeviceResolverDistributed(dev_mgr, &wc_, worker_name); in DefineWorker()
175 new CollectiveParamResolverDistributed(config, dev_mgr, dev_res, &wc_, in DefineWorker()
180 wc_.AddWorker(worker_name, fw); in DefineWorker()
283 FakeCache wc_; member in tensorflow::__anon53767e020111::DeviceResDistTest
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma_mgr.cc | 155 rdma_adapter_->wc_); in ConnectivityCheck()
161 ibv_wc_status s = rdma_adapter_->wc_[i].status; in ConnectivityCheck()
163 if ((int)rdma_adapter_->wc_[i].wr_id == RdmaChannel::kPingRecvWrid) { in ConnectivityCheck()
165 << ": " << ibv_wc_status_str(rdma_adapter_->wc_[i].status) << "(" in ConnectivityCheck()
166 << rdma_adapter_->wc_[i].status << ") for PING_RECV_WRID"; in ConnectivityCheck()
171 reinterpret_cast<RdmaChannel*>(rdma_adapter_->wc_[i].wr_id); in ConnectivityCheck()
173 << ": " << ibv_wc_status_str(rdma_adapter_->wc_[i].status) << "(" in ConnectivityCheck()
174 << rdma_adapter_->wc_[i].status << ") to " << rc->remote_name_; in ConnectivityCheck()
|
| D | rdma.cc | 447 ibv_poll_cq(cq_, MAX_CONCURRENT_WRITES * 2, static_cast<ibv_wc*>(wc_)); in Process_CQ()
450 CHECK(wc_[i].status == IBV_WC_SUCCESS) in Process_CQ()
452 << ibv_wc_status_str(wc_[i].status) << " " << wc_[i].status << " " in Process_CQ()
453 << static_cast<int>(wc_[i].wr_id) << " " << wc_[i].vendor_err; in Process_CQ()
454 if (wc_[i].opcode == IBV_WC_RECV_RDMA_WITH_IMM) { in Process_CQ()
455 RdmaChannel* rc = reinterpret_cast<RdmaChannel*>(wc_[i].wr_id); in Process_CQ()
459 uint32_t imm_data = wc_[i].imm_data; in Process_CQ()
504 } else if (wc_[i].opcode == IBV_WC_RDMA_WRITE) { in Process_CQ()
505 RdmaWriteID* wr_id = reinterpret_cast<RdmaWriteID*>(wc_[i].wr_id); in Process_CQ()
|
| D | rdma.h | 398 ibv_wc wc_[MAX_CONCURRENT_WRITES * 2]; variable
|