Searched refs:stra5 (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
34 self.transpose2 = P.Transpose().shard(stra5)
73 stra5 = ((8, 1, 1, 1),)
75 … net = Net(_wi, _wo, stra1=stra1, stra2=stra2, stra3=stra3, stra4=stra4, stra5=stra5, stra6=stra6)
86 stra5 = ((4, 1, 1, 1),)
88 … net = Net(_wi, _wo, stra1=stra1, stra2=stra2, stra3=stra3, stra4=stra4, stra5=stra5, stra6=stra6)
|