Searched refs:MatMulInfo (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/ |
| D | edge_costmodel_test.cc | 26 using MatMulInfoPtr = std::shared_ptr<MatMulInfo>;
68 matmul1 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_1, outputs_shape_1, attr_1); in SetUp()
77 matmul2 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_2, outputs_shape_2, attr_2); in SetUp()
86 matmul3 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_3, outputs_shape_3, attr_3); in SetUp()
95 matmul4 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_4, outputs_shape_4, attr_4); in SetUp()
104 matmul5 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_5, outputs_shape_5, attr_5); in SetUp()
|
| D | dp_algo_test.cc | 28 using MatMulInfoPtr = std::shared_ptr<MatMulInfo>;
178 matmul0 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_0, outputs_shape_0, attr_0); in SetUp()
188 matmul1 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_1, outputs_shape_1, attr_1); in SetUp()
198 matmul2 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_2, outputs_shape_2, attr_2); in SetUp()
208 matmul3 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_3, outputs_shape_3, attr_3); in SetUp()
218 matmul4 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_4, outputs_shape_4, attr_4); in SetUp()
228 matmul5 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_5, outputs_shape_5, attr_5); in SetUp()
238 matmul6 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_6, outputs_shape_6, attr_6); in SetUp()
248 matmul7 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_7, outputs_shape_7, attr_7); in SetUp()
258 matmul8 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_8, outputs_shape_8, attr_8); in SetUp()
[all …]
|
| D | graph_costmodel_test.cc | 25 using MatMulInfoPtr = std::shared_ptr<MatMulInfo>;
77 matmul0 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_0, outputs_shape_0, attr_0); in SetUp()
86 matmul1 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_1, outputs_shape_1, attr_1); in SetUp()
95 matmul2 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_2, outputs_shape_2, attr_2); in SetUp()
104 matmul3 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_3, outputs_shape_3, attr_3); in SetUp()
113 matmul4 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_4, outputs_shape_4, attr_4); in SetUp()
122 matmul5 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_5, outputs_shape_5, attr_5); in SetUp()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | matmul_info.h | 81 class MatMulInfo : public MatMul {
83 MatMulInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in MatMulInfo() function
86 ~MatMulInfo() override = default;
|
| /third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/ |
| D | construct_operator_test.cc | 28 class MatMulInfo;
29 using MatMulInfoPtr = std::shared_ptr<MatMulInfo>;
63 …MatMulInfoPtr matmul = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_1, outputs_shape_1… in SetUp()
|
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | matmul_info_test.cc | 30 class MatMulInfo;
31 using MatMulInfoPtr = std::shared_ptr<MatMulInfo>;
69 matmul1 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_1, outputs_shape_1, attr_1); in SetUp()
79 matmul2 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_2, outputs_shape_2, attr_2); in SetUp()
89 matmul3 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_3, outputs_shape_3, attr_3); in SetUp()
93 matmul4 = std::make_shared<MatMulInfo>("matmul_info", inputs_shape_3, outputs_shape_3, attr_4); in SetUp()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | dynamic_creator.h | 79 REGISTER(MatMulInfo);
|