| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 61 string NcclCollectiveKey(const string& exec_key, int step_id) { in NcclCollectiveKey() argument
62 return strings::StrCat(exec_key, ":", step_id); in NcclCollectiveKey()
77 NcclCollectiveKey(col_ctx->exec_key, col_ctx->step_id); in Enqueue()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_alg.cc | 63 string RingAlgBufKey(const string& name, const string& exec_key, int pass, in RingAlgBufKey() argument
66 return strings::StrCat(name, "(", exec_key, "):pass(", pass, "):section(", in RingAlgBufKey()
71 return strings::StrCat(exec_key, ":", pass, ":", section, ":", source_rank); in RingAlgBufKey()
384 string send_buf_key = RingAlgBufKey(name_, col_ctx_->exec_key, in DispatchSend()
404 RingAlgBufKey(name_, col_ctx_->exec_key, rf->second_pass, rf->sc_idx, in DispatchRecv()
425 " exec ", col_ctx_->exec_key, " step_id=", col_ctx_->step_id, in FieldState()
|
| D | hierarchical_tree_broadcaster.cc | 44 string BroadcastBufKey(const string& exec_key, int subdiv, int src_rank, in BroadcastBufKey() argument
47 return strings::StrCat("broadcast(", exec_key, "):subdiv(", subdiv, in BroadcastBufKey()
51 return strings::StrCat(exec_key, ":", subdiv, ":", src_rank, ":", dst_rank); in BroadcastBufKey()
414 BroadcastBufKey(col_ctx_->exec_key, subdiv, src_rank, dst_rank); in DispatchSend()
434 BroadcastBufKey(col_ctx_->exec_key, subdiv, src_rank, dst_rank); in DispatchRecv()
|
| D | permuter.cc | 83 strings::StrCat(col_ctx_->exec_key, src_rank, target_rank); in DispatchSend()
99 strings::StrCat(col_ctx_->exec_key, src_rank, target_rank); in DispatchRecv()
|
| D | base_collective_executor.h | 114 const string& exec_key, StatusCallback done) override;
|
| D | base_collective_executor.cc | 268 const string& exec_key, in ExecuteAsync() argument
317 col_params, exec_key, step_id_, input, output); in ExecuteAsync()
|
| D | permuter_test.cc | 397 string exec_key = in DoPermute() local
403 parent_->dev_mgr_.get(), &ctx, &op_params, col_params_, exec_key, in DoPermute()
|
| D | ring_gatherer_test.cc | 486 string exec_key = in DoGather() local
492 parent_->dev_mgr_.get(), &ctx, &op_params, col_params_, exec_key, in DoGather()
|
| D | ring_reducer_test.cc | 519 string exec_key = in DoReduce() local
525 parent_->dev_mgr_.get(), &ctx, &op_params, col_params_, exec_key, in DoReduce()
|
| D | hierarchical_tree_broadcaster_test.cc | 679 string exec_key = in DoBroadcast() local
686 parent_->dev_mgr_.get(), &ctx, &op_params, col_params_, exec_key, in DoBroadcast()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 173 const string& exec_key, int64 step_id, const Tensor* input, Tensor* output) in CollectiveContext() argument
180 exec_key(exec_key), in CollectiveContext()
|
| D | collective.h | 302 const string& exec_key, StatusCallback done) { in ExecuteAsync() argument
371 const string exec_key; member
383 const CollectiveParams* col_params, const string& exec_key,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 326 string exec_key = in RunReduce() local
332 /*OpKernelContext=*/&ctx, &op_params, col_params_, exec_key, kStepId, in RunReduce()
358 string exec_key = in RunBroadcast() local
364 /*OpKernelContext=*/&ctx, &op_params, col_params_, exec_key, kStepId, in RunBroadcast()
399 string exec_key = in RunGather() local
405 /*OpKernelContext=*/&ctx, &op_params, col_params_, exec_key, kStepId, in RunGather()
|