Searched refs:col_exec (Results 1 – 9 of 9) 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->collective_executor()->RunClosure([this, c, done, col_exec]() { in CanProceedWithCompute()
50 col_exec->CompleteParamsAsync( in CanProceedWithCompute()
94 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync() local
96 c, col_exec, in ComputeAsync()
117 if (!CanProceedWithCompute(c, col_exec, done)) return; in ComputeAsync()
130 col_exec->ExecuteAsync(c, col_params_, GetCollectiveKey(c), actual_done); in ComputeAsync()
218 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync() local
220 c, col_exec, in ComputeAsync()
238 if (!CanProceedWithCompute(c, col_exec, done)) return; in ComputeAsync()
[all …]
|
| D | collective_nccl_gatherer.cc | 55 col_ctx_->col_exec->WaitForDependencies(*col_params_); in Run()
64 col_ctx_->col_exec->UnblockDependencies(*col_params_); in Run()
|
| D | collective_nccl_broadcaster.cc | 65 col_ctx_->col_exec->WaitForDependencies(*col_params_); in Run()
74 col_ctx_->col_exec->UnblockDependencies(*col_params_); in Run()
|
| D | collective_nccl_reducer.cc | 176 col_ctx_->col_exec->WaitForDependencies(*col_params_); in Run()
185 col_ctx_->col_exec->UnblockDependencies(*col_params_); in Run()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 161 CollectiveContext::CollectiveContext(CollectiveExecutor* col_exec, in CollectiveContext() argument
168 : col_exec(col_exec), in CollectiveContext()
|
| D | collective.h | 357 CollectiveContext(CollectiveExecutor* col_exec, const DeviceMgr* dev_mgr,
364 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 | 420 col_ctx_->col_exec->PostToPeer(col_params_->instance.device_names[dst_idx], in DispatchSend()
439 col_ctx_->col_exec->RecvFromPeer( in DispatchRecv()
|
| D | ring_reducer.cc | 61 col_ctx_->col_exec->UnblockDependencies(*col_params_); in Run()
|