Searched refs:stra6 (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_batchmm.py | 26 stra5=None, stra6=None): argument
35 self.relu2 = P.ReLU().shard(stra6)
74 stra6 = ((8, 1, 1, 1),)
75 … net = Net(_wi, _wo, stra1=stra1, stra2=stra2, stra3=stra3, stra4=stra4, stra5=stra5, stra6=stra6)
87 stra6 = ((4, 1, 1, 1),)
88 … net = Net(_wi, _wo, stra1=stra1, stra2=stra2, stra3=stra3, stra4=stra4, stra5=stra5, stra6=stra6)
|