Searched defs:src_rank (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | permuter.cc | 80 void Permuter::DispatchSend(int src_rank, int target_rank, const Tensor* tensor, in DispatchSend()
96 void Permuter::DispatchRecv(int src_rank, int target_rank, Tensor* tensor, in DispatchRecv()
|
| D | all_to_all.cc | 123 void AllToAll::DispatchSend(int src_rank, int target_rank, const Tensor* tensor, in DispatchSend()
135 void AllToAll::DispatchRecv(int src_rank, int target_rank, Tensor* tensor, in DispatchRecv()
|
| D | hierarchical_tree_broadcaster.cc | 45 string BroadcastBufKey(const string& exec_key, int subdiv, int src_rank, in BroadcastBufKey()
408 int src_rank, in DispatchSend()
433 void HierarchicalTreeBroadcaster::DispatchRecv(int subdiv, int src_rank, in DispatchRecv()
|
| D | eval_const_tensor.cc | 87 int src_rank = shape_c->Rank(shape_c->input(0)); in TryToInferTensorOutputFromShapeNode() local
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 211 TensorShape shape, int src_node, int src_rank, in MakeBroadcastTestCase()
624 const int src_rank = static_cast<int>(random::New64() % num_ranks); in TYPED_TEST() local
|