| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_stridedslice.py | 65 _w2 = Tensor(np.ones([128, 64, 1]), dtype=ms.float32) variable
82 … net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), (1, 1, 1), strategy1, strategy2, is_parameter=True)
91 … net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), (1, 1, 2), strategy1, strategy2, is_parameter=True)
100 …net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), (1, 1, 1), strategy1, strategy2, is_parameter=True, …
109 net = Net(_w1, _w2, (0, 0), (128, 64), (1, 1), strategy1, strategy2, is_parameter=True)
117 … net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), (1, 1, 1), strategy1, strategy2, is_parameter=True)
125 … net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), (1, 1, 1), strategy1, strategy2, is_parameter=False)
133 … net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), (1, 1, 1), strategy1, strategy2, is_parameter=True)
141 net = Net2(_w2, (0, 0, 0), (64, 64, 1), (1, 1, 1), strategy1, strategy2)
149 net = Net2(_w2, (0, 0, 0), (64, 64, 1), (1, 1, 1), strategy1, strategy2)
[all …]
|
| D | test_o2_level.py | 89 _w2 = Tensor(np.ones([128, 64, 1]), dtype=ms.float32) variable
108 net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), (1, 1, 1),
119 net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), (1, 1, 2),
130 net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), (1, 1, 1),
141 net = Net(_w1, _w2, (0, 0), (128, 64), (1, 1),
151 net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), (1, 1, 1),
161 net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), (1, 1, 1),
171 net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), (1, 1, 1),
181 net = Net2(_w2, (0, 0, 0), (64, 64, 1), (1, 1, 1), strategy1, strategy2)
190 net = Net2(_w2, (0, 0, 0), (64, 64, 1), (1, 1, 1), strategy1, strategy2)
[all …]
|
| D | test_slice.py | 63 _w2 = Tensor(np.ones([128, 64, 1]), dtype=ms.float32) variable
80 net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), strategy1, strategy2, is_parameter=True)
88 net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), strategy1, strategy2)
96 net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), strategy1, strategy2, is_parameter=False)
104 net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), strategy1, strategy2, is_parameter=True)
112 net = Net2(_w2, (0, 0, 0), (64, 64, 1), strategy1, strategy2)
120 net = Net2(_w2, (0, 0, 0), (64, 64, 1), strategy1, strategy2)
128 net = Net2(_w2, (0, 0, 0), (128, 64, 1), strategy1, strategy2)
134 net = Net2(_w2, (0, 0, 0), (32, 64, 1))
|
| D | test_batch_matmul.py | 40 _w2 = Tensor(np.ones([128, 32, 32]), dtype=ms.float32) variable
57 net = Net(_w1, _w2, False, strategy1, strategy2)
65 net = Net(_w1, _w2, False, strategy1, strategy2)
73 net = Net(_w1, _w2, False, strategy1, strategy2)
79 net = Net(_w1, _w2, False)
87 net = Net(_w1, _w2, False, strategy1, strategy2)
95 net = Net(_w1, _w2, True, strategy1, strategy2)
|
| D | test_tile.py | 75 _w2 = Tensor(np.ones([128, 64, 32]), dtype=ms.float32) variable
93 net = Net(_w1, _w2, strategy1, strategy2, is_parameter=True)
101 net = Net(_w1, _w2, strategy1, strategy2, is_parameter=True)
109 net = Net(_w1, _w2, strategy1, strategy2, is_parameter=False)
117 net = Net(_w1, _w2, strategy1, strategy2, is_parameter=False)
133 net = Net2(_w2, strategy1, strategy2)
141 net = Net2(_w2, strategy1, strategy2)
149 net = Net2(_w2, strategy1, strategy2)
155 net = Net2(_w2)
|
| D | test_pack.py | 130 _w2 = Tensor(np.ones([48, 64]), dtype=ms.float32) variable
181 net = Net(_w1, _w2, 0, strategy1, strategy2)
189 net = Net(_w1, _w2, 0, strategy1, strategy2)
197 net = Net(_w1, _w2, 0, strategy1, strategy2, False)
205 net = Net1(_w1, _w2, 0, strategy1, strategy2)
213 net = Net1(_w1, _w2, 1, strategy1, strategy2)
221 net = Net1(_w1, _w2, 0, strategy1, strategy2)
229 net = Net1(_w1, _w2, 0, strategy1, strategy2)
237 net = Net1(_w1, _w2, 1, strategy1, strategy2)
243 net = Net1(_w1, _w2, 0)
[all …]
|
| D | test_select.py | 64 _w2 = Tensor(np.ones([128, 64, 32]), dtype=ms.float32) variable
83 net = Net(_w1, _w2, strategy1, strategy2)
92 net = Net(_w1, _w2, strategy1, strategy2)
101 net = Net(_w1, _w2, strategy1, strategy2)
108 net = Net(_w1, _w2)
117 net = Net(_w1, _w2, strategy1, strategy2)
|
| D | test_conv2d.py | 44 _w2 = Tensor(np.ones([8, 16, 3, 3]), dtype=ms.float32) variable
135 …net = Net(_w2, out_channel=8, kernel_size=3, pad_mode="same", stride=1, strategy1=strategy1, strat…
141 net = Net(_w2, out_channel=8, kernel_size=3, pad_mode="same", stride=1)
149 …net = Net(_w2, out_channel=8, kernel_size=3, pad_mode="same", stride=1, strategy1=strategy1, strat…
157 …net = Net(_w2, out_channel=8, kernel_size=3, pad_mode="same", stride=2, strategy1=strategy1, strat…
166 …net = Net(_w2, out_channel=8, kernel_size=3, pad_mode="same", stride=1, strategy1=strategy1, strat…
175 …net = Net(_w2, out_channel=8, kernel_size=3, pad_mode="valid", stride=1, strategy1=strategy1, stra…
274 …net = Net(_w2, out_channel=8, kernel_size=3, pad_mode="same", stride=1, strategy1=strategy1, strat…
|
| D | test_split.py | 75 _w2 = Tensor(np.ones([48, 64, 32]), dtype=ms.float32) variable
126 net = Net2(_w2, 0, 2, strategy1, strategy2)
134 net = Net2(_w2, 0, 2, strategy1, strategy2)
142 net = Net2(_w2, 0, 2, strategy1, strategy2)
148 net = Net2(_w2, 0, 2)
|
| D | test_concat.py | 73 _w2 = Tensor(np.ones([32, 64, 32]), dtype=ms.float32) variable
95 net = Net(_w1, _w2, strategy1, strategy2, is_parameter=True)
103 net = Net(_w1, _w2, strategy1, strategy2, is_parameter=True)
111 net = Net(_w1, _w2, strategy1, strategy2, is_parameter=False)
141 net = Net2(_w2)
|
| D | test_model_without_loss.py | 97 _w2 = Tensor(np.ones([32, 64, 32]), dtype=ms.float32) variable
121 net = Net(_w1, _w2, strategy1, strategy2, is_parameter=True)
130 net = Net(_w1, _w2, strategy1, strategy2, is_parameter=True)
139 net = Net(_w1, _w2, strategy1, strategy2, is_parameter=False)
173 net = Net2(_w2)
|
| D | test_adafactor.py | 41 _w2 = Tensor(np.ones([32]), dtype=ms.float32) variable
64 net = Net(_w1, _w2, strategy1, strategy2)
72 net = Net(_w1, _w2, strategy1, strategy2)
|
| D | test_conv2d_transpose.py | 57 _w2 = Tensor(np.ones([8, 16, 4, 4]), dtype=ms.float32) variable
92 net = Net2(_w2, out_channel=8, kernel_size=(4, 4), pad_mode="same", stride=2,
101 net = Net2(_w2, out_channel=8, kernel_size=(4, 4), pad_mode="same", stride=2,
149 net = Net2(_w2, out_channel=8, kernel_size=(4, 4), pad_mode="same", stride=2,
|
| D | test_parameter_merge.py | 71 _w2 = Tensor(np.ones([128, 64, 1]), dtype=ms.float32) variable
102 net = Net(_w1, _w2, (0, 0, 0), (128, 64, 32), (1, 1, 1),
|