Searched defs:dev_mat (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/ |
| D | _tensor.py | 28 def _get_tensor_strategy(dev_mat, tensor_map): argument
178 def _get_slice_index(dev_mat, tensor_map, opt_shard_group): argument
201 def _load_tensor(tensor, dev_mat, tensor_map, full_shape=None, rank_id=-1): argument
278 def _reshape_param_data(param_data, dev_mat, tensor_map): argument
628 def _reshape_param_data_with_weight(param_data, dev_mat, field_size): argument
|
| /third_party/mindspore/mindspore-src/source/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-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | virtual_dataset_info.cc | 144 const Shape &dev_mat) { in ObtainTensorMap()
|
| D | operator_info.cc | 1132 Shape dev_mat = {repeated_size, device_num / repeated_size}; in CreateGroupForOptShard() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_layout.cc | 632 Shape dev_mat(device_arrangement_origin_.array()); in TransferRepeatLayout() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_grad_experimental/ |
| D | grad_comm_ops.py | 667 def bprop(x, dev_mat, tensor_map, out, dout): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | comm_ops.py | 1782 def infer_value(self, x, dev_mat, tensor_map, slice_shape, full_shape): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | graph_utils.cc | 1297 auto dev_mat = out_layout.device_arrangement(); in SegmentEntireShapeToPartialForDynamic() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | parameter_manager.cc | 1481 auto dev_mat = tensor_layout->device_arrangement().array(); in HandleCameAndAdaFactorOpt() local
|