| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | onehot_info_test_axis_0.cc | 72 Shape dev_matrix_shape = onehot_info2->dev_matrix_shape(); in TEST_F() local
84 Shape dev_matrix_shape = onehot_info2->dev_matrix_shape(); in TEST_F() local
96 Shape dev_matrix_shape = onehot_info2->dev_matrix_shape(); in TEST_F() local
|
| D | onehot_info_test.cc | 72 Shape dev_matrix_shape = onehot_info->dev_matrix_shape(); in TEST_F() local
84 Shape dev_matrix_shape = onehot_info->dev_matrix_shape(); in TEST_F() local
96 Shape dev_matrix_shape = onehot_info->dev_matrix_shape(); in TEST_F() local
|
| D | matmul_info_test.cc | 101 Shape dev_matrix_shape = matmul1->dev_matrix_shape(); in TEST_F() local
112 Shape dev_matrix_shape = matmul1->dev_matrix_shape(); in TEST_F() local
124 Shape dev_matrix_shape = matmul2->dev_matrix_shape(); in TEST_F() local
136 Shape dev_matrix_shape = matmul2->dev_matrix_shape(); in TEST_F() local
148 Shape dev_matrix_shape = matmul3->dev_matrix_shape(); in TEST_F() local
160 Shape dev_matrix_shape = matmul3->dev_matrix_shape(); in TEST_F() local
|
| D | transpose_test.cc | 75 Shape dev_matrix_shape = transpose->dev_matrix_shape(); in TEST_F() local
86 Shape dev_matrix_shape = transpose->dev_matrix_shape(); in TEST_F() local
|
| D | reshape_test.cc | 75 Shape dev_matrix_shape = reshape->dev_matrix_shape(); in TEST_F() local
86 Shape dev_matrix_shape = reshape->dev_matrix_shape(); in TEST_F() local
|
| D | prelu_test.cc | 71 Shape dev_matrix_shape = prelu->dev_matrix_shape(); in TEST_F() local
176 Shape dev_matrix_shape = prelu_2d->dev_matrix_shape(); in TEST_F() local
|
| D | tanh_info_test.cc | 70 Shape dev_matrix_shape = tanh->dev_matrix_shape(); in TEST_F() local
|
| D | pow_info_test.cc | 70 Shape dev_matrix_shape = pow->dev_matrix_shape(); in TEST_F() local
|
| D | gelu_info_test.cc | 70 Shape dev_matrix_shape = gelu->dev_matrix_shape(); in TEST_F() local
|
| D | tmpidentity_test.cc | 72 Shape dev_matrix_shape = identity_ptr->dev_matrix_shape(); in TEST_F() local
|
| D | activation_info_test.cc | 71 Shape dev_matrix_shape = activation->dev_matrix_shape(); in TEST_F() local
|
| D | log_softmax_info_test.cc | 71 Shape dev_matrix_shape = log_softmax->dev_matrix_shape(); in TEST_F() local
|
| D | l2_normalize_info_test.cc | 71 Shape dev_matrix_shape = norm->dev_matrix_shape(); in TEST_F() local
|
| D | softmax_info_test.cc | 75 Shape dev_matrix_shape = softmax->dev_matrix_shape(); in TEST_F() local
|
| D | softmax_entropy_loss_info_test.cc | 71 Shape dev_matrix_shape = loss->dev_matrix_shape(); in TEST_F() local
|
| D | reduce_method_test.cc | 75 Shape dev_matrix_shape = reduce_sum->dev_matrix_shape(); in TEST_F() local
|
| D | tensor_add_info_test.cc | 73 Shape dev_matrix_shape = tensor_add->dev_matrix_shape(); in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | arithmetic_info.cc | 113 TensorMap SetExpendTensorMap(const Shape &strategy, const Shape &dev_matrix_shape) { in SetExpendTensorMap()
125 TensorMap SetTensorMap(const Shape &strategy_expend, const Shape &dev_matrix_shape, const Shape &st… in SetTensorMap()
|
| D | matmul_info.cc | 36 Shape *dev_matrix_shape) { in SetDevMatrixShape()
|
| D | operator_info.h | 121 Shape dev_matrix_shape() const { return dev_matrix_shape_; } in dev_matrix_shape() function
|
| D | operator_info.cc | 461 Shape dev_matrix_shape = tensor_layout.device_arrangement().array(); in CreateGetTensorSliceOp() local
1564 int64_t ComputeRepeatDeviceNumByTensorMap(const Shape &dev_matrix_shape, const Shape &tensor_map) { in ComputeRepeatDeviceNumByTensorMap()
|
| /third_party/mindspore/tests/ut/cpp/parallel/ |
| D | virtual_dataset_test.cc | 68 Shape dev_matrix_shape = virtual_dataset->dev_matrix_shape(); in TEST_F() local
|
| /third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/ |
| D | construct_operator_test.cc | 69 Shape dev_matrix_shape = {2, 4, 8, 16, 1}; in SetUp() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | construct_operator.cc | 25 Status ConstructOperator::Init(const RankList &dev_list, const Shape &dev_matrix_shape) { in Init()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | parameter_manager.cc | 203 Shape dev_matrix_shape = tensor_info.tensor_layout().device_arrangement().array(); in GetGroupByTensorInfo() local
|