| /third_party/mindspore/mindspore-src/source/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 | 110 Shape dev_matrix_shape = matmul1->dev_matrix_shape(); in TEST_F() local
124 Shape dev_matrix_shape = matmul1->dev_matrix_shape(); in TEST_F() local
138 Shape dev_matrix_shape = matmul2->dev_matrix_shape(); in TEST_F() local
152 Shape dev_matrix_shape = matmul2->dev_matrix_shape(); in TEST_F() local
166 Shape dev_matrix_shape = matmul3->dev_matrix_shape(); in TEST_F() local
180 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 | gelu_info_test.cc | 70 Shape dev_matrix_shape = gelu->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 | 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 | softmax_entropy_loss_info_test.cc | 71 Shape dev_matrix_shape = loss->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 | reduce_method_test.cc | 76 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-src/source/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/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | quant_info.cc | 486 TensorMap SetExpandQuantTensorMap(const Shape &strategy, const Shape &dev_matrix_shape) { in SetExpandQuantTensorMap()
498 TensorMap SetQuantTensorMap(const Shape &strategy_expand, const Shape &dev_matrix_shape, const Shap… in SetQuantTensorMap()
|
| D | arithmetic_info.cc | 196 TensorMap SetExpandTensorMap(const Shape &strategy, const Shape &dev_matrix_shape) { in SetExpandTensorMap()
208 TensorMap SetTensorMap(const Shape &strategy_expand, const Shape &dev_matrix_shape, const Shape &st… in SetTensorMap()
|
| D | gather_info.h | 73 Shape dev_matrix_shape() const { return dev_matrix_shape_; } in dev_matrix_shape() function
|
| D | operator_info.h | 199 Shape dev_matrix_shape() const { return dev_matrix_shape_; } in dev_matrix_shape() function
|
| D | flash_attention_score_info.cc | 864 auto dev_matrix_shape = dev_matrix_shape_; in InferSplitNumAndDevMatrixShapeByLayout() local
|
| /third_party/mindspore/mindspore-src/source/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-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | construct_operator.cc | 28 Status ConstructOperator::Init(const RankList &dev_list, const Shape &dev_matrix_shape, bool is_cos… in Init()
|