Searched defs:src_rank (Results 1 – 4 of 4) sorted by relevance
80 int64_t src_rank = src_shape.size(); in Init() local
226 bool HcomUtil::GetHcomSrcRank(const AnfNodePtr &anf_node, uint32_t *src_rank) { in GetHcomSrcRank()
477 def __init__(self, sr_tag, src_rank, shape, dtype, group=GlobalComm.WORLD_COMM_GROUP, argument
624 auto src_rank = global_rank_ - (user_node_stage - node_stage) * per_stage_rank_num_; in InsertReceive() local