Searched refs:cem_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 336 return cem_->GetParamResolver()->CompleteGroupAsync(device, group_params, in CompleteGroupAsync()
341 return cem_->GetParamResolver()->LookupGroup(group_key, group); in LookupGroup()
382 : cem_(cem) {} in CollectiveExecutor()
386 CollectiveExecutorMgrInterface* cem_; variable
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 250 cem_->GetParamResolver()->StartAbort(status); in StartAbort()
252 if (cem_->GetNcclCommunicator() != nullptr) { in StartAbort()
253 cem_->GetNcclCommunicator()->StartAbort(status); in StartAbort()
323 this, cem_->GetNcclCommunicator(), dev_mgr_, ctx, CtxParams(ctx), in ExecuteAsync()
401 cem_->GetParamResolver()->CompleteParamsAsync(device, cp, cancel_mgr, in CompleteParamsAsync()
|