Searched refs:_w3 (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_concat.py | 74 _w3 = Tensor(np.ones([128, 16, 32]), dtype=ms.float32) variable
135 net = Net2(_w3, strategy1, strategy2, axis=1)
149 net = Net2(_w3, strategy1, strategy2, axis=1)
|
| D | test_model_without_loss.py | 98 _w3 = Tensor(np.ones([128, 16, 32]), dtype=ms.float32) variable
166 net = Net2(_w3, strategy1, strategy2, axis=1)
182 net = Net2(_w3, strategy1, strategy2, axis=1)
|
| D | test_conv2d.py | 45 _w3 = Tensor(np.ones([8, 16, 5, 5]), dtype=ms.float32) variable
256 …net = Net(_w3, out_channel=8, kernel_size=5, pad_mode="same", stride=3, strategy1=strategy1, strat…
265 …net = Net(_w3, out_channel=8, kernel_size=5, pad_mode="same", stride=1, strategy1=strategy1, strat…
|
| D | test_conv2d_transpose.py | 58 _w3 = Tensor(np.ones([8, 16, 10, 10]), dtype=ms.float32) variable
139 net = Net2(_w3, out_channel=8, kernel_size=(10, 10), pad_mode="same", stride=2,
|
| D | test_pack.py | 131 _w3 = Tensor(np.ones([48, 64]), dtype=ms.float32) variable
255 net = Net2(_w1, _w2, _w3)
|
| D | test_tile.py | 76 _w3 = Tensor(np.ones([128, 16, 16]), dtype=ms.float32) variable
|