Searched refs:col_exec (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_ops.cc | 38 bool CanProceedWithCompute(OpKernelContext* c, CollectiveExecutor* col_exec, in CanProceedWithCompute() argument
45 c->env()->SchedClosure([this, c, done, col_exec]() { in CanProceedWithCompute()
50 col_exec->CompleteParamsAsync( in CanProceedWithCompute()
88 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync() local
90 c, col_exec, in ComputeAsync()
105 if (!CanProceedWithCompute(c, col_exec, done)) return; in ComputeAsync()
110 col_exec->ExecuteAsync(c, col_params_, GetCollectiveKey(c), actual_done); in ComputeAsync()
186 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync() local
188 c, col_exec, in ComputeAsync()
206 if (!CanProceedWithCompute(c, col_exec, done)) return; in ComputeAsync()
[all …]
|
| D | collective_nccl_reducer.cc | 181 col_ctx_->col_exec->WaitForDependencies(*col_params_); in Run()
187 col_ctx_->col_exec->Launched(*col_params_); in Run()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 156 CollectiveContext::CollectiveContext(CollectiveExecutor* col_exec, in CollectiveContext() argument
163 : col_exec(col_exec), in CollectiveContext()
|
| D | collective.h | 341 CollectiveContext(CollectiveExecutor* col_exec, const DeviceMgr* dev_mgr,
348 CollectiveExecutor* col_exec; // Not owned variable
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_alg.cc | 289 col_ctx_->col_exec->StartAbort(s); in StartAbort()
390 col_ctx_->col_exec->PostToPeer( in DispatchSend()
409 col_ctx_->col_exec->RecvFromPeer( in DispatchRecv()
|
| D | hierarchical_tree_broadcaster.cc | 410 col_ctx_->col_exec->PostToPeer(col_params_->instance.device_names[dst_idx], in DispatchSend()
429 col_ctx_->col_exec->RecvFromPeer( in DispatchRecv()
|