Searched defs:dev_mat (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/parallel/ |
| D | _tensor.py | 22 def _get_tensor_strategy(dev_mat, tensor_map): argument
172 def _get_slice_index(dev_mat, tensor_map): argument
189 def _load_tensor(tensor, dev_mat, tensor_map): argument
252 def _reshape_param_data(param_data, dev_mat, tensor_map): argument
306 def _reshape_param_data_with_weight(param_data, dev_mat, field_size): argument
|
| /third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/ |
| D | redistribution_operator_infer_test.cc | 73 Shape dev_mat; in InferOperatorCheckAll() local
|
| D | construct_operator_test.cc | 85 Arrangement dev_mat; in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_layout.cc | 409 Shape dev_mat(device_arrangement_origin_.array()); in TransferRepeatLayout() local
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_comm_ops.py | 534 def bprop(x, dev_mat, tensor_map, out, dout): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | parameter_manager.cc | 583 auto dev_mat = tensor_layout->device_arrangement().array(); in HandleAdaFactorOpt() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | comm_ops.py | 927 def infer_value(self, x, dev_mat, tensor_map): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | operator_info.cc | 588 Shape dev_mat = {repeated_size, device_num / repeated_size}; in CreateGroupForOptShard() local
|