Searched refs:stra1 (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_batchmm.py | 25 def __init__(self, wi, wo, stra1=None, stra2=None, stra3=None, stra4=None, argument
28 self.relu = P.ReLU().shard(stra1)
69 stra1 = ((8, 1, 1, 1),)
75 … net = Net(_wi, _wo, stra1=stra1, stra2=stra2, stra3=stra3, stra4=stra4, stra5=stra5, stra6=stra6)
82 stra1 = ((4, 1, 1, 1),)
88 … net = Net(_wi, _wo, stra1=stra1, stra2=stra2, stra3=stra3, stra4=stra4, stra5=stra5, stra6=stra6)
|
| D | test_onehot_2dim.py | 25 def __init__(self, wi, stra1=None, stra2=None, stra3=None): argument
28 self.matmul = P.MatMul().shard(stra1)
60 stra1 = ((8, 1), (1, 1))
63 net = Net(_wi, stra1=stra1, stra2=stra2, stra3=stra3)
|
| /third_party/mindspore/tests/ut/cpp/parallel/ |
| D | strategy_test.cc | 76 StrategyPtr stra1 = std::make_shared<Strategy>(stage1, inputs1); in TEST_F() local
81 ASSERT_EQ(stra1->IsEqual(stra2), true); in TEST_F()
82 ASSERT_EQ(stra1->IsEqual(stra3), false); in TEST_F()
83 ASSERT_EQ(stra1->IsEqual(stra4), false); in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | prelu_test.cc | 165 auto stra1 = stra->GetInputDim()[1]; in TEST_F() local
167 ASSERT_EQ(stra1[0], 1); in TEST_F()
270 auto stra1 = stra->GetInputDim()[1]; in TEST_F() local
272 ASSERT_EQ(stra1[0], 1); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | get_next_info.cc | 90 … [](const Dimensions &stra1, const Dimensions &stra2) { return stra1.size() < stra2.size(); }); in InferDevMatrixShape() argument
|