Searched defs:exec_key (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 61 string NcclCollectiveKey(const string& exec_key, int step_id) { in NcclCollectiveKey()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 326 string exec_key = in RunReduce() local
358 string exec_key = in RunBroadcast() local
399 string exec_key = in RunGather() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 302 const string& exec_key, StatusCallback done) { in ExecuteAsync()
371 const string exec_key; member
|
| D | collective.cc | 173 const string& exec_key, int64 step_id, const Tensor* input, Tensor* output) in CollectiveContext()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_alg.cc | 63 string RingAlgBufKey(const string& name, const string& exec_key, int pass, in RingAlgBufKey()
|
| D | hierarchical_tree_broadcaster.cc | 44 string BroadcastBufKey(const string& exec_key, int subdiv, int src_rank, in BroadcastBufKey()
|
| D | base_collective_executor.cc | 268 const string& exec_key, in ExecuteAsync()
|
| D | permuter_test.cc | 397 string exec_key = in DoPermute() local
|
| D | ring_gatherer_test.cc | 486 string exec_key = in DoGather() local
|
| D | ring_reducer_test.cc | 519 string exec_key = in DoReduce() local
|
| D | hierarchical_tree_broadcaster_test.cc | 679 string exec_key = in DoBroadcast() local
|