| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_alg.cc | 62 string RingAlgBufKey(const string& name, const string& exec_key, int pass, in RingAlgBufKey() argument
65 return strings::StrCat(name, "(", exec_key, "):pass(", pass, "):section(", in RingAlgBufKey()
70 return strings::StrCat(exec_key, ":", pass, ":", section, ":", source_rank); in RingAlgBufKey()
382 string send_buf_key = RingAlgBufKey(name_, col_ctx_->exec_key, in DispatchSend()
401 RingAlgBufKey(name_, col_ctx_->exec_key, rf->second_pass, rf->sc_idx, in DispatchRecv()
421 " exec ", col_ctx_->exec_key, " step_id=", col_ctx_->step_id, in FieldState()
|
| D | hierarchical_tree_broadcaster.cc | 43 string BroadcastBufKey(const string& exec_key, int subdiv, int src_rank, in BroadcastBufKey() argument
46 return strings::StrCat("broadcast(", exec_key, "):subdiv(", subdiv, in BroadcastBufKey()
50 return strings::StrCat(exec_key, ":", subdiv, ":", src_rank, ":", dst_rank); in BroadcastBufKey()
403 BroadcastBufKey(col_ctx_->exec_key, subdiv, src_rank, dst_rank); in DispatchSend()
422 BroadcastBufKey(col_ctx_->exec_key, subdiv, src_rank, dst_rank); in DispatchRecv()
|
| D | base_collective_executor.cc | 220 const string& exec_key, in ExecuteAsync() argument
255 exec_key, step_id_, input, output); in ExecuteAsync()
|
| D | base_collective_executor.h | 111 const string& exec_key, StatusCallback done) override;
|
| D | ring_gatherer_test.cc | 480 string exec_key = in DoGather() local
484 &ctx, &op_params, col_params_, exec_key, in DoGather()
|
| D | ring_reducer_test.cc | 510 string exec_key = in DoReduce() local
514 &ctx, &op_params, col_params_, exec_key, in DoReduce()
|
| D | hierarchical_tree_broadcaster_test.cc | 681 string exec_key = in DoBroadcast() local
685 &ctx, &op_params, col_params_, exec_key, in DoBroadcast()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer.cc | 24 string NcclCollectiveKey(const string& exec_key, int step_id) { in NcclCollectiveKey() argument
25 return strings::StrCat(exec_key, ":", step_id); in NcclCollectiveKey()
139 NcclCollectiveKey(col_ctx_->exec_key, col_ctx_->step_id); in Run()
|
| D | collective_nccl_reducer_test.cc | 288 string exec_key = in DoReduce() local
292 &ctx, &op_params, col_params_, exec_key, in DoReduce()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 161 const string& exec_key, int64 step_id, in CollectiveContext() argument
168 exec_key(exec_key), in CollectiveContext()
|
| D | collective.h | 263 const string& exec_key, StatusCallback done) { in ExecuteAsync() argument
343 const CollectiveParams& col_params, const string& exec_key,
353 const string exec_key; variable
|