Searched refs:GetDevicesAlongDim (Results 1 – 6 of 6) sorted by relevance
47 Status status = GetDevicesAlongDim(SizeToUlong(i), &group); in CreateGroupList()56 Status DeviceMatrix::GetDevicesAlongDim(const uint64_t &dim, RankList *devices) { in GetDevicesAlongDim() function in mindspore::parallel::DeviceMatrix
40 Status GetDevicesAlongDim(const uint64_t &dim, RankList *devices);
255 if (dev_matrix.GetDevicesAlongDim(SizeToUlong(axis), &group_devices) != SUCCESS) { in CreateGroupByDim()
591 if (temp_dev_matrix.GetDevicesAlongDim(0, &mirror_group_devices) != SUCCESS) { in CreateGroupForOptShard()627 if (dev_matrix.GetDevicesAlongDim(SizeToUlong(axis), &group_devices) != SUCCESS) { in CreateGroupByDim()
697 if (dev_matrix.GetDevicesAlongDim(SizeToUlong(dim), &group_devices) != SUCCESS) { in InferGroup()
379 if (dev_matrix.GetDevicesAlongDim(w_index_in_dev_matrix, &group_devices) != SUCCESS) { in InferRankBias()