Home
last modified time | relevance | path

Searched defs:exec_key (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/nccl/
Dcollective_communicator.cc61 string NcclCollectiveKey(const string& exec_key, int step_id) { in NcclCollectiveKey()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_test.cc326 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/
Dcollective.h302 const string& exec_key, StatusCallback done) { in ExecuteAsync()
371 const string exec_key; member
Dcollective.cc173 const string& exec_key, int64 step_id, const Tensor* input, Tensor* output) in CollectiveContext()
/external/tensorflow/tensorflow/core/common_runtime/
Dring_alg.cc63 string RingAlgBufKey(const string& name, const string& exec_key, int pass, in RingAlgBufKey()
Dhierarchical_tree_broadcaster.cc44 string BroadcastBufKey(const string& exec_key, int subdiv, int src_rank, in BroadcastBufKey()
Dbase_collective_executor.cc268 const string& exec_key, in ExecuteAsync()
Dpermuter_test.cc397 string exec_key = in DoPermute() local
Dring_gatherer_test.cc486 string exec_key = in DoGather() local
Dring_reducer_test.cc519 string exec_key = in DoReduce() local
Dhierarchical_tree_broadcaster_test.cc679 string exec_key = in DoBroadcast() local