Home
last modified time | relevance | path

Searched refs:dev_matrix_shape (Results 1 – 25 of 26) sorted by relevance

12

/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Donehot_info_test_axis_0.cc72 Shape dev_matrix_shape = onehot_info2->dev_matrix_shape(); in TEST_F() local
75 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
84 Shape dev_matrix_shape = onehot_info2->dev_matrix_shape(); in TEST_F() local
87 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
96 Shape dev_matrix_shape = onehot_info2->dev_matrix_shape(); in TEST_F() local
99 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Donehot_info_test.cc72 Shape dev_matrix_shape = onehot_info->dev_matrix_shape(); in TEST_F() local
75 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
84 Shape dev_matrix_shape = onehot_info->dev_matrix_shape(); in TEST_F() local
87 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
96 Shape dev_matrix_shape = onehot_info->dev_matrix_shape(); in TEST_F() local
99 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Dmatmul_info_test.cc101 Shape dev_matrix_shape = matmul1->dev_matrix_shape(); in TEST_F() local
104 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
112 Shape dev_matrix_shape = matmul1->dev_matrix_shape(); in TEST_F() local
115 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
124 Shape dev_matrix_shape = matmul2->dev_matrix_shape(); in TEST_F() local
127 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
136 Shape dev_matrix_shape = matmul2->dev_matrix_shape(); in TEST_F() local
139 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
148 Shape dev_matrix_shape = matmul3->dev_matrix_shape(); in TEST_F() local
151 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
[all …]
Dtranspose_test.cc75 Shape dev_matrix_shape = transpose->dev_matrix_shape(); in TEST_F() local
78 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
86 Shape dev_matrix_shape = transpose->dev_matrix_shape(); in TEST_F() local
89 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Dreshape_test.cc75 Shape dev_matrix_shape = reshape->dev_matrix_shape(); in TEST_F() local
78 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
86 Shape dev_matrix_shape = reshape->dev_matrix_shape(); in TEST_F() local
89 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Dprelu_test.cc71 Shape dev_matrix_shape = prelu->dev_matrix_shape(); in TEST_F() local
74 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
176 Shape dev_matrix_shape = prelu_2d->dev_matrix_shape(); in TEST_F() local
179 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Dgelu_info_test.cc70 Shape dev_matrix_shape = gelu->dev_matrix_shape(); in TEST_F() local
73 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Dlog_softmax_info_test.cc71 Shape dev_matrix_shape = log_softmax->dev_matrix_shape(); in TEST_F() local
74 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Dpow_info_test.cc70 Shape dev_matrix_shape = pow->dev_matrix_shape(); in TEST_F() local
73 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Dtanh_info_test.cc70 Shape dev_matrix_shape = tanh->dev_matrix_shape(); in TEST_F() local
73 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Dsoftmax_info_test.cc75 Shape dev_matrix_shape = softmax->dev_matrix_shape(); in TEST_F() local
78 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Dactivation_info_test.cc71 Shape dev_matrix_shape = activation->dev_matrix_shape(); in TEST_F() local
74 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Dsoftmax_entropy_loss_info_test.cc71 Shape dev_matrix_shape = loss->dev_matrix_shape(); in TEST_F() local
74 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Dtmpidentity_test.cc72 Shape dev_matrix_shape = identity_ptr->dev_matrix_shape(); in TEST_F() local
75 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Dl2_normalize_info_test.cc71 Shape dev_matrix_shape = norm->dev_matrix_shape(); in TEST_F() local
74 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Dreduce_method_test.cc75 Shape dev_matrix_shape = reduce_sum->dev_matrix_shape(); in TEST_F() local
78 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
Dtensor_add_info_test.cc73 Shape dev_matrix_shape = tensor_add->dev_matrix_shape(); in TEST_F() local
76 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
/third_party/mindspore/tests/ut/cpp/parallel/
Dvirtual_dataset_test.cc68 Shape dev_matrix_shape = virtual_dataset->dev_matrix_shape(); in TEST_F() local
71 ASSERT_EQ(dev_matrix_shape, expect); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Darithmetic_info.cc113 TensorMap SetExpendTensorMap(const Shape &strategy, const Shape &dev_matrix_shape) { in SetExpendTensorMap() argument
116 if (strategy[i] == dev_matrix_shape[i]) { in SetExpendTensorMap()
125 TensorMap SetTensorMap(const Shape &strategy_expend, const Shape &dev_matrix_shape, const Shape &st… in SetTensorMap() argument
126 TensorMap expend_map = SetExpendTensorMap(strategy_expend, dev_matrix_shape); in SetTensorMap()
127 size_t dev_matrix_size = dev_matrix_shape.size(); in SetTensorMap()
Dmatmul_info.cc36 Shape *dev_matrix_shape) { in SetDevMatrixShape() argument
37 MS_EXCEPTION_IF_NULL(dev_matrix_shape); in SetDevMatrixShape()
46 dev_matrix_shape->push_back(mat_a_strategy.at(i)); in SetDevMatrixShape()
54 dev_matrix_shape->push_back(mat_b_strategy.at(i)); in SetDevMatrixShape()
59 dev_matrix_shape->push_back(mat_a_strategy.at(SECOND_FROM_END(mat_a_size))); in SetDevMatrixShape()
60 dev_matrix_shape->push_back(mat_a_strategy.back()); in SetDevMatrixShape()
64 dev_matrix_shape->push_back(mat_b_strategy.back()); in SetDevMatrixShape()
66 dev_matrix_shape->push_back(mat_b_strategy.at(SECOND_FROM_END(mat_b_size))); in SetDevMatrixShape()
Doperator_info.cc461 Shape dev_matrix_shape = tensor_layout.device_arrangement().array(); in CreateGetTensorSliceOp() local
465 ValuePtr dev_mat_value = MakeValue(dev_matrix_shape); in CreateGetTensorSliceOp()
474 << ShapeToString(dev_matrix_shape) << ", " << ShapeToString(tensor_map); in CreateGetTensorSliceOp()
1564 int64_t ComputeRepeatDeviceNumByTensorMap(const Shape &dev_matrix_shape, const Shape &tensor_map) { in ComputeRepeatDeviceNumByTensorMap() argument
1569 …int64_t device_num = std::accumulate(dev_matrix_shape.begin(), dev_matrix_shape.end(), 1, std::mul… in ComputeRepeatDeviceNumByTensorMap()
1574 } else if ((element < 0) || (LongToSize(element) >= dev_matrix_shape.size())) { in ComputeRepeatDeviceNumByTensorMap()
1576 << ShapeToString(dev_matrix_shape); in ComputeRepeatDeviceNumByTensorMap()
1579 size_t index = dev_matrix_shape.size() - LongToSize(element) - 1; in ComputeRepeatDeviceNumByTensorMap()
1580 if (dev_matrix_shape[index] <= 0) { in ComputeRepeatDeviceNumByTensorMap()
1581 MS_LOG(ERROR) << "Invalid dev matrix shape: " << ShapeToString(dev_matrix_shape); in ComputeRepeatDeviceNumByTensorMap()
[all …]
Doperator_info.h121 Shape dev_matrix_shape() const { return dev_matrix_shape_; } in dev_matrix_shape() function
312 int64_t ComputeRepeatDeviceNumByTensorMap(const Shape &dev_matrix_shape, const Shape &tensor_map);
/third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/
Dconstruct_operator_test.cc69 Shape dev_matrix_shape = {2, 4, 8, 16, 1}; in SetUp() local
71 constructor.Init(used_dev_list, dev_matrix_shape); in SetUp()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dconstruct_operator.cc25 Status ConstructOperator::Init(const RankList &dev_list, const Shape &dev_matrix_shape) { in Init() argument
26 dev_size_ = dev_matrix_shape.size(); in Init()
27 dev_matrix_shape_ = dev_matrix_shape; in Init()
Dconstruct_operator.h37 Status Init(const RankList &dev_list, const Shape &dev_matrix_shape);

12