Searched refs:dev_matrix (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/ |
| D | parallel_strategy_checkpoint.cc | 169 straspb::DevMatrix *dev_matrix = parallel_layouts->add_dev_matrix(); in Save() local
170 MS_EXCEPTION_IF_NULL(dev_matrix); in Save()
172 dev_matrix->add_dim(UlongToUint(LongToUlong(dev_dim))); in Save()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | reshape_info.cc | 197 Arrangement dev_matrix; in InferTensorLayout() local
198 Status status = dev_matrix.Init(dev_matrix_shape_); in InferTensorLayout()
216 (void)tensor_layout_in.Init(dev_matrix, tensor_map_in, shape_in); in InferTensorLayout()
233 (void)tensor_layout_out.Init(dev_matrix, tensor_map_out, shape_out); in InferTensorLayout()
|
| D | operator_info.cc | 531 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in CreateGroupByTensorMap() local
533 if (dev_matrix.GetDevicesByTensorMap(tensor_map, &group_devices) != SUCCESS) { in CreateGroupByTensorMap()
554 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in CreateGroupForOptShard() local
557 if (dev_matrix.GetDevicesByTensorMap(tensor_map, &group_devices) != SUCCESS) { in CreateGroupForOptShard()
625 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in CreateGroupByDim() local
627 if (dev_matrix.GetDevicesAlongDim(SizeToUlong(axis), &group_devices) != SUCCESS) { in CreateGroupByDim()
|
| D | gather_v2_p_info.cc | 684 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in InferGroup() local
697 if (dev_matrix.GetDevicesAlongDim(SizeToUlong(dim), &group_devices) != SUCCESS) { in InferGroup()
|
| D | conv2d_info.cc | 377 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in InferRankBias() local
379 if (dev_matrix.GetDevicesAlongDim(w_index_in_dev_matrix, &group_devices) != SUCCESS) { in InferRankBias()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | parameter_manager.cc | 206 DeviceMatrix dev_matrix(rank, stage_device_list, dev_matrix_shape); in GetGroupByTensorInfo() local
208 if (dev_matrix.GetDevicesByTensorMap(tensor_map, &group_devices) != SUCCESS) { in GetGroupByTensorInfo()
387 DeviceMatrix dev_matrix(rank, rank_list, dev_mat_shape); in IsFullySplitParameter() local
389 if (dev_matrix.GetDevicesByTensorMap(tensor_map, &group_devices) != SUCCESS) { in IsFullySplitParameter()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | node_strategy.proto | 52 repeated DevMatrix dev_matrix = 1; field
|
| /third_party/mindspore/mindspore/train/ |
| D | serialization.py | 1110 dev_mat = list(layout.dev_matrix[0].dim)
1468 dev_matrix, tensor_map, param_split_shape, field_size = predict_strategy[key][:4]
1469 if not _check_int_list(dev_matrix) or not _check_int_list(tensor_map) or \
1495 dev_mat = list(layout.dev_matrix[0].dim)
1514 dev_matrix = layout.dev_matrix.add()
1516 dev_matrix.dim.append(item)
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | construct_operator.cc | 253 DeviceMatrix dev_matrix(rank, dev_list_, dev_matrix_shape_); in CreateGroupByDim() local
255 if (dev_matrix.GetDevicesAlongDim(SizeToUlong(axis), &group_devices) != SUCCESS) { in CreateGroupByDim()
|