Searched defs:src_rank (Results 1 – 4 of 4) 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 | hierarchical_tree_broadcaster.cc | 44 string BroadcastBufKey(const string& exec_key, int subdiv, int src_rank, in BroadcastBufKey()
407 int src_rank, in DispatchSend()
430 void HierarchicalTreeBroadcaster::DispatchRecv(int subdiv, int src_rank, in DispatchRecv()
|
| /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
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 1314 int src_rank = src_type.getRank(); in matchAndRewrite() local
|