Searched refs:GetDevicesByTensorMap (Results 1 – 5 of 5) sorted by relevance
92 arr.GetDevicesByTensorMap(tensor_map, &rank_list); in TEST_F()103 arr.GetDevicesByTensorMap(tensor_map, &rank_list); in TEST_F()114 arr.GetDevicesByTensorMap(tensor_map, &rank_list); in TEST_F()
39 Status GetDevicesByTensorMap(const Shape &tensor_map, RankList *rank_list);
118 Status DeviceMatrix::GetDevicesByTensorMap(const Shape &tensor_map, RankList *rank_list) { in GetDevicesByTensorMap() function in mindspore::parallel::DeviceMatrix
208 if (dev_matrix.GetDevicesByTensorMap(tensor_map, &group_devices) != SUCCESS) { in GetGroupByTensorInfo()389 if (dev_matrix.GetDevicesByTensorMap(tensor_map, &group_devices) != SUCCESS) { in IsFullySplitParameter()
533 if (dev_matrix.GetDevicesByTensorMap(tensor_map, &group_devices) != SUCCESS) { in CreateGroupByTensorMap()557 if (dev_matrix.GetDevicesByTensorMap(tensor_map, &group_devices) != SUCCESS) { in CreateGroupForOptShard()