Searched defs:dev_matrix (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | node_strategy.proto | 52 repeated DevMatrix dev_matrix = 1; field
|
| /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
|
| /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
387 DeviceMatrix dev_matrix(rank, rank_list, dev_mat_shape); in IsFullySplitParameter() local
|
| /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
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | reshape_info.cc | 197 Arrangement dev_matrix; in InferTensorLayout() local
|
| D | operator_info.cc | 531 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in CreateGroupByTensorMap() local
554 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in CreateGroupForOptShard() local
625 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in CreateGroupByDim() local
|
| D | gather_v2_p_info.cc | 684 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in InferGroup() local
|
| D | conv2d_info.cc | 377 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in InferRankBias() local
|