Searched refs:target_rank (Results 1 – 2 of 2) sorted by relevance
334 int target_rank = send_to_ranks[i]; in RunTree() local339 DispatchSend(si, target_rank, my_rank, in RunTree()
379 def _right_pad_with_ones(tensor, target_rank): argument387 [target_rank - array_ops.rank(tensor)],388 dtype=target_rank.dtype)