Home
last modified time | relevance | path

Searched defs:src_rank (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dindex_add_gpu_kernel.h80 int64_t src_rank = src_shape.size(); in Init() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhcom_util.cc226 bool HcomUtil::GetHcomSrcRank(const AnfNodePtr &anf_node, uint32_t *src_rank) { in GetHcomSrcRank()
/third_party/mindspore/mindspore/ops/operations/
D_inner_ops.py477 def __init__(self, sr_tag, src_rank, shape, dtype, group=GlobalComm.WORLD_COMM_GROUP, argument
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc624 auto src_rank = global_rank_ - (user_node_stage - node_stage) * per_stage_rank_num_; in InsertReceive() local