Searched refs:GetCollectiveKey (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | nccl_ops.cc | 50 string GetCollectiveKey(OpKernelContext* c) { in GetCollectiveKey() function in tensorflow::__anonc894cd950111::NcclAsyncOpBase
115 {GetCollectiveKey(c), in ComputeAsync()
146 {GetCollectiveKey(c), in ComputeAsync()
182 {GetCollectiveKey(c), in ComputeAsync()
215 std::move(participant), {GetCollectiveKey(c), in ComputeAsync()
251 std::move(participant), {GetCollectiveKey(c), in ComputeAsync()
|
| D | collective_ops.cc | 92 string GetCollectiveKey(OpKernelContext* c) { in GetCollectiveKey() function in tensorflow::__anon09108d7f0111::CollectiveOpV1Kernel
197 col_exec->ExecuteAsync(c, col_params_, GetCollectiveKey(c), actual_done); in ComputeAsyncImpl()
305 col_exec->ExecuteAsync(c, col_params_, GetCollectiveKey(c), actual_done); in ComputeAsyncImpl()
384 col_exec->ExecuteAsync(c, col_params_, GetCollectiveKey(c), actual_done); in ComputeAsyncImpl()
454 col_exec->ExecuteAsync(c, col_params_, GetCollectiveKey(c), actual_done); in ComputeAsyncImpl()
|