Home
last modified time | relevance | path

Searched refs:CreateGroupByDim (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dconstruct_operator.cc99 if (CreateGroupByDim(dev_size_ - LongToSize(dev_dim) - 1, &group_list) != SUCCESS) { in StridedSliceOP()
146 if (CreateGroupByDim(dev_size_ - LongToSize(dev_dim) - 1, &group_list) != SUCCESS) { in AllGatherOP()
224 if (CreateGroupByDim(dev_size_ - LongToSize(dev_dim) - 1, &group_list) != SUCCESS) { in AlltoAllOP()
248 Status ConstructOperator::CreateGroupByDim(size_t axis, std::vector<Group> *group) { in CreateGroupByDim() function in mindspore::parallel::ConstructOperator
Dconstruct_operator.h54 Status CreateGroupByDim(size_t axis, std::vector<Group> *group);
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.h204 Status CreateGroupByDim(size_t axis, std::vector<Group> *group);
Dmatmul_info.cc228 if (CreateGroupByDim(relevant_dimension_index, &group_list) != SUCCESS) { in InferForwardCommunication()
Dconv2d_info.cc514 if (CreateGroupByDim(relevant_dim_index, &group_list) != SUCCESS) { in InferForwardCommunication()
Doperator_info.cc618 Status OperatorInfo::CreateGroupByDim(size_t axis, std::vector<Group> *group) { in CreateGroupByDim() function in mindspore::parallel::OperatorInfo