| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_gather_v2.py | 40 class GradWrap(nn.Cell): class
42 super(GradWrap, self).__init__()
69 net = GradWrap(NetWithLoss(Net(0, strategy1, strategy2)))
82 net = GradWrap(NetWithLoss(Net(0, strategy1, strategy2)))
95 net = GradWrap(NetWithLoss(Net(0, strategy1, strategy2)))
108 net = GradWrap(NetWithLoss(Net(1, strategy1, strategy2)))
121 net = GradWrap(NetWithLoss(Net(1, strategy1, strategy2)))
134 net = GradWrap(NetWithLoss(Net(1, strategy1, strategy2)))
146 net = GradWrap(NetWithLoss(Net(0, None, strategy2)))
158 net = GradWrap(NetWithLoss(Net(1, None, strategy2)))
[all …]
|
| D | test_sparse_gather_v2.py | 41 class GradWrap(nn.Cell): class
43 super(GradWrap, self).__init__()
70 net = GradWrap(NetWithLoss(Net(0, strategy1, strategy2)))
83 net = GradWrap(NetWithLoss(Net(0, strategy1, strategy2)))
96 net = GradWrap(NetWithLoss(Net(0, strategy1, strategy2)))
109 net = GradWrap(NetWithLoss(Net(1, strategy1, strategy2)))
122 net = GradWrap(NetWithLoss(Net(1, strategy1, strategy2)))
135 net = GradWrap(NetWithLoss(Net(1, strategy1, strategy2)))
147 net = GradWrap(NetWithLoss(Net(0, None, strategy2)))
159 net = GradWrap(NetWithLoss(Net(1, None, strategy2)))
[all …]
|
| D | test_gpu_dropout.py | 40 class GradWrap(nn.Cell): class
42 super(GradWrap, self).__init__()
63 net = GradWrap(NetWithLoss(Net()))
76 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
89 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
100 net = GradWrap(NetWithLoss(Net()))
|
| D | test_comparison_function_info.py | 41 class GradWrap(nn.Cell): class
43 super(GradWrap, self).__init__()
71 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
94 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
118 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
142 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
166 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
190 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
214 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
237 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
[all …]
|
| D | test_reshape_unexpand.py | 42 class GradWrap(nn.Cell): class
44 super(GradWrap, self).__init__()
67 net = GradWrap(NetWithLoss(Net()))
90 net = GradWrap(NetWithLoss(Net()))
113 net = GradWrap(NetWithLoss(Net()))
137 net = GradWrap(NetWithLoss(Net()))
161 net = GradWrap(NetWithLoss(Net()))
185 net = GradWrap(NetWithLoss(Net()))
209 net = GradWrap(NetWithLoss(Net()))
244 net = GradWrap(NetWithLoss(Net()))
[all …]
|
| D | test_self_attention.py | 44 class GradWrap(nn.Cell): class
46 super(GradWrap, self).__init__()
98 net = GradWrap(NetWithLoss(
116 net = GradWrap(NetWithLoss(
134 net = GradWrap(NetWithLoss(
145 net = GradWrap(NetWithLoss(
|
| D | test_virtual_dataset_with_strategy.py | 39 class GradWrap(nn.Cell): class
41 super(GradWrap, self).__init__()
92 net = GradWrap(NetWithLoss(Net1(strategy1, strategy2, strategy3)))
106 net = GradWrap(NetWithLoss(Net1(strategy1, strategy2, strategy3)))
123 net = GradWrap(NetWithLoss(Net2(strategy1, strategy2, strategy3)))
137 net = GradWrap(NetWithLoss(Net2(strategy1, strategy2, strategy3)))
151 net = GradWrap(NetWithLoss(Net2(strategy1, strategy2, strategy3)))
165 net = GradWrap(NetWithLoss(Net2(strategy1, strategy2, strategy3)))
|
| D | test_prelu.py | 41 class GradWrap(nn.Cell): class
43 super(GradWrap, self).__init__()
67 net = GradWrap(NetWithLoss(Net()))
84 net = GradWrap(NetWithLoss(Net()))
106 net = GradWrap(NetWithLoss(Net(strategy)))
126 net = GradWrap(NetWithLoss(Net(strategy)))
190 net = GradWrap(NetWithLoss(Net(strategy)))
210 net = GradWrap(NetWithLoss(Net(strategy)))
|
| D | test_element_wise_function.py | 41 class GradWrap(nn.Cell): class
43 super(GradWrap, self).__init__()
73 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
99 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
125 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
150 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
175 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
201 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
226 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
252 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
[all …]
|
| D | test_embeddinglookup.py | 29 class GradWrap(nn.Cell): class
31 super(GradWrap, self).__init__()
88 net = GradWrap(NetWithLoss(Net(shape, offset)))
100 net = GradWrap(NetWithLoss(Net(shape, offset)))
115 net = GradWrap(NetWithLoss(Net(shape, offset, strategy1, strategy2, "CPU")))
130 net = GradWrap(NetWithLoss(Net(shape, offset, strategy1, strategy2, "CPU")))
|
| D | test_different_type_for_div_op.py | 29 class GradWrap(nn.Cell): class
31 super(GradWrap, self).__init__()
59 net = GradWrap(Net(strategy0, strategy1))
82 net = GradWrap(Net(strategy0, strategy1))
105 net = GradWrap(Net(strategy0, strategy1))
|
| D | test_strategy_checkpoint.py | 43 class GradWrap(nn.Cell): class
45 super(GradWrap, self).__init__()
86 … net = GradWrap(NetWithLoss(Net(strategy1, strategy2, strategy3, strategy4, strategy5, strategy6)))
107 class GradWrap(nn.Cell): class
109 super(GradWrap, self).__init__()
149 … net = GradWrap(NetWithLoss(Net(strategy1, strategy3, strategy4, strategy5, strategy6, strategy7)))
171 class GradWrap(nn.Cell): class
173 super(GradWrap, self).__init__()
207 net = GradWrap(NetWithLoss(Net()))
228 class GradWrap(nn.Cell): class
[all …]
|
| D | test_arithmetic.py | 40 class GradWrap(nn.Cell): class
42 super(GradWrap, self).__init__()
71 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
95 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
119 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
142 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
165 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
189 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
213 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
242 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
[all …]
|
| D | test_reshape_parameter.py | 41 class GradWrap(nn.Cell): class
43 super(GradWrap, self).__init__()
73 net = GradWrap(NetWithLoss(Net(strategy)))
82 net = GradWrap(NetWithLoss(Net(strategy)))
|
| D | test_two_matmul.py | 41 class GradWrap(nn.Cell): class
43 super(GradWrap, self).__init__()
72 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
97 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
121 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
146 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
170 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
|
| D | test_sum_as_loss.py | 29 class GradWrap(nn.Cell): class
31 super(GradWrap, self).__init__()
59 net = GradWrap(Net(strategy0, strategy1))
82 net = GradWrap(Net(strategy0, strategy1))
|
| D | test_auto_parallel_reduce_method.py | 41 class GradWrap(nn.Cell): class
43 super(GradWrap, self).__init__()
72 net = GradWrap(NetWithLoss(Net()))
96 net = GradWrap(NetWithLoss(Net()))
120 net = GradWrap(NetWithLoss(Net()))
|
| D | test_softmax_cross_entropy_loss.py | 40 class GradWrap(nn.Cell): class
42 super(GradWrap, self).__init__()
71 net = GradWrap(NetWithLoss(Net(strategy1, strategy2), strategy3))
96 net = GradWrap(NetWithLoss(Net(strategy1, strategy2), strategy3))
118 net = GradWrap(NetWithLoss(Net()))
|
| D | test_get_next.py | 48 class GradWrap(nn.Cell): class
50 super(GradWrap, self).__init__()
76 net = GradWrap(NetWithLoss(Net(), [ms.float32, ms.int32], [[32, 64], [32]], 2))
100 net = GradWrap(net_with_loss)
124 net = GradWrap(net_with_loss)
145 net = GradWrap(net_with_loss)
|
| D | test_matmul_tensor.py | 43 class GradWrap(nn.Cell): class
45 super(GradWrap, self).__init__()
80 net = GradWrap(NetWithLoss(Net(strategy1, strategy2, strategy3)))
105 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
134 net = GradWrap(NetWithLoss(Net(strategy1, strategy2, strategy3)))
159 net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
|
| D | test_hybird_parallel_activation.py | 41 class GradWrap(nn.Cell): class
43 super(GradWrap, self).__init__()
72 net = GradWrap(NetWithLoss(Net(strategy1, strategy2, strategy3)))
98 net = GradWrap(NetWithLoss(Net(strategy1, strategy2, strategy3)))
124 net = GradWrap(NetWithLoss(Net(strategy1, strategy2, strategy3)))
150 net = GradWrap(NetWithLoss(Net(strategy1, strategy2, strategy3)))
179 net = GradWrap(NetWithLoss(Net(strategy1, strategy2, strategy3)))
211 … net = GradWrap(NetWithLoss(Net(strategy1, strategy2, strategy3, strategy4, strategy5, strategy6)))
243 … net = GradWrap(NetWithLoss(Net(strategy1, strategy2, strategy3, strategy4, strategy5, strategy6)))
|
| D | test_scalar_loss.py | 30 class GradWrap(nn.Cell): class
32 super(GradWrap, self).__init__()
57 net = GradWrap(Net(strategy0, strategy1))
|
| /third_party/mindspore/tests/ut/python/pynative_mode/ |
| D | test_sparse_pynative.py | 35 class GradWrap(nn.Cell): class
37 super(GradWrap, self).__init__()
55 GradWrap(RowTensorGetAttr((3, 2)))(indices, values)
70 GradWrap(SparseTensorGetAttr())(indices, values)
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_minimum_grad.py | 37 class GradWrap(Cell): class
39 super(GradWrap, self).__init__()
63 net_me = GradWrap(MinNetMe())
|
| D | test_maximum_grad.py | 37 class GradWrap(Cell): class
39 super(GradWrap, self).__init__()
61 net_me = GradWrap(MaxNetMe())
|