Searched refs:target_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() argument
83 strings::StrCat(col_ctx_->exec_key, src_rank, target_rank); in DispatchSend()
86 << col_params_->instance.devices[target_rank] in DispatchSend()
87 << " target_rank=" << target_rank << " src_rank=" << src_rank; in DispatchSend()
89 col_params_->instance.devices[target_rank], in DispatchSend()
90 col_params_->group.task_names[target_rank], send_buf_key, in DispatchSend()
96 void Permuter::DispatchRecv(int src_rank, int target_rank, Tensor* tensor, in DispatchRecv() argument
99 strings::StrCat(col_ctx_->exec_key, src_rank, target_rank); in DispatchRecv()
103 << " target_rank=" << target_rank << " src_rank=" << src_rank; in DispatchRecv()
|
| D | permuter.h | 74 void DispatchSend(int src_rank, int target_rank, const Tensor* tensor,
77 void DispatchRecv(int src_rank, int target_rank, Tensor* tensor,
|
| D | hierarchical_tree_broadcaster.cc | 341 int target_rank = send_to_ranks[i]; in RunTree() local
346 DispatchSend(si, target_rank, my_rank, in RunTree()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 309 int64 target_rank = target_sharding.tile_assignment().num_dimensions() - in PartialReplicateReshardCompatibleSharding() local
311 if (target_rank != rank) { in PartialReplicateReshardCompatibleSharding()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 5053 target_rank = target.shape.ndims
5056 target_rank is not None and output_rank is not None and
5057 target_rank != output_rank - 1)
|