Searched refs:col_exec (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_ops.cc | 61 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync() local
63 c, col_exec, in ComputeAsync()
71 !c->cancellation_manager()->RegisterCallback(token, [col_exec]() { in ComputeAsync()
75 col_exec->StartAbort(errors::Cancelled("op cancelled")); in ComputeAsync()
80 auto deregister_and_done = [c, col_exec, token, done = std::move(done)]() { in ComputeAsync()
87 ComputeAsyncImpl(c, col_exec, std::move(deregister_and_done)); in ComputeAsync()
99 bool CanProceedWithCompute(OpKernelContext* c, CollectiveExecutor* col_exec, in CanProceedWithCompute() argument
106 c->collective_executor()->RunClosure([this, c, col_exec, done]() { in CanProceedWithCompute()
111 col_exec->CompleteParamsAsync( in CanProceedWithCompute()
130 CollectiveExecutor* col_exec,
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | permuter.cc | 88 col_ctx_->col_exec->remote_access()->PostToPeer( in DispatchSend()
104 col_ctx_->col_exec->remote_access()->RecvFromPeer( in DispatchRecv()
|
| D | ring_alg.cc | 290 col_ctx_->col_exec->StartAbort(s); in StartAbort()
392 col_ctx_->col_exec->remote_access()->PostToPeer( in DispatchSend()
412 col_ctx_->col_exec->remote_access()->RecvFromPeer( in DispatchRecv()
|
| D | hierarchical_tree_broadcaster.cc | 421 col_ctx_->col_exec->remote_access()->PostToPeer( in DispatchSend()
441 col_ctx_->col_exec->remote_access()->RecvFromPeer( in DispatchRecv()
|
| D | ring_reducer.cc | 60 col_ctx_->col_exec->UnblockDependencies(*col_params_); in Run()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 170 CollectiveExecutor* col_exec, NcclCommunicatorInterface* nccl_communicator, in CollectiveContext() argument
174 : col_exec(col_exec), in CollectiveContext()
|
| D | collective.h | 365 CollectiveExecutor* col_exec; // Not owned member
379 CollectiveContext(CollectiveExecutor* col_exec,
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 188 col_ctx->col_exec->WaitForDependencies(*col_params); in Enqueue()
197 col_ctx->col_exec->UnblockDependencies(*col_params); in Enqueue()
|