Searched refs:outputs_dims (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/parallel/ |
| D | step_parallel_test.cc | 240 Shape outputs_dims = {64, 64}; in TEST_F() local
241 CNodePtr node = Make_Node(inputs_x_dims, inputs_y_dims, outputs_dims, 4); in TEST_F()
248 Shape outputs_dims = {64, 64}; in TEST_F() local
249 CNodePtr node = Make_Node(inputs_x_dims, inputs_y_dims, outputs_dims); in TEST_F()
252 Shapes outputs_shape = std::vector<Shape>{outputs_dims}; in TEST_F()
260 Shape outputs_dims = {64, 64}; in TEST_F() local
261 CNodePtr node = Make_Node(inputs_x_dims, inputs_y_dims, outputs_dims, 1); in TEST_F()
268 Shape outputs_dims = {64, 64}; in TEST_F() local
269 CNodePtr node = Make_Node(inputs_x_dims, inputs_y_dims, outputs_dims, 3); in TEST_F()
496 Shape outputs_dims = {64, 64}; in TEST_F() local
[all …]
|
| D | step_auto_parallel_test.cc | 141 Shape outputs_dims = {64, 64}; in TEST_F() local
142 CNodePtr node = Create_Node(inputs_x_dims, inputs_y_dims, outputs_dims); in TEST_F()
151 Shapes outputs_shape = std::vector<Shape>{outputs_dims}; in TEST_F()
|