Searched refs:CreateGroupForOptShard (Results 1 – 3 of 3) sorted by relevance
184 Status CreateGroupForOptShard(TensorLayout *const tensor_layout, std::vector<Group> *group);
547 Status OperatorInfo::CreateGroupForOptShard(TensorLayout *const tensor_layout, std::vector<Group> *… in CreateGroupForOptShard() function in mindspore::parallel::OperatorInfo
1624 if (distribute_operator->CreateGroupForOptShard(tensor_layout, &dev_group) == Status::SUCCESS && in GetOptShardGroup()