Home
last modified time | relevance | path

Searched refs:GradNet (Results 1 – 25 of 57) sorted by relevance

123

/third_party/mindspore/tests/ut/python/optimizer/
Dtest_bprop_mindir.py50 class GradNet(nn.Cell): class
52 super(GradNet, self).__init__()
70 grad = GradNet(relu)
90 grad = GradNet(relu)
97 grad = GradNet(identity)
104 grad = GradNet(range_net)
111 grad = GradNet(ones_like)
118 grad = GradNet(zeros_like)
125 grad = GradNet(argmax)
132 grad = GradNet(argmin)
[all …]
/third_party/mindspore/tests/ut/python/parameter_feature/
Dtest_var_grad.py107 class GradNet(Cell): class
109 super(GradNet, self).__init__()
120 grad_net = GradNet(net)
125 class GradNet(Cell): class
127 super(GradNet, self).__init__()
137 grad_net = GradNet(net)
142 class GradNet(Cell): class
144 super(GradNet, self).__init__()
154 grad_net = GradNet(net)
159 class GradNet(Cell): class
[all …]
/third_party/mindspore/tests/ut/python/pynative_mode/
Dtest_user_define_bprop_check.py40 class GradNet(nn.Cell): class
42 super(GradNet, self).__init__()
52 grad_net = GradNet(net)
72 class GradNet(nn.Cell): class
74 super(GradNet, self).__init__()
84 grad_net = GradNet(net)
104 class GradNet(nn.Cell): class
106 super(GradNet, self).__init__()
116 grad_net = GradNet(net)
134 class GradNet(nn.Cell): class
[all …]
Dtest_implicit_conversion.py141 class GradNet(nn.Cell): class
143 super(GradNet, self).__init__()
153 grad_net = GradNet(net)
169 class GradNet(nn.Cell): class
171 super(GradNet, self).__init__()
181 grad_net = GradNet(net)
197 class GradNet(nn.Cell): class
199 super(GradNet, self).__init__()
209 grad_net = GradNet(net)
225 class GradNet(nn.Cell): class
[all …]
Dtest_cont_cases.py89 class GradNet(nn.Cell): class
91 super(GradNet, self).__init__()
99 net = GradNet(while_net)
151 class GradNet(nn.Cell): class
153 super(GradNet, self).__init__()
162 net = GradNet(while_net)
191 class GradNet(nn.Cell): class
193 super(GradNet, self).__init__()
202 net = GradNet(while_net)
228 class GradNet(nn.Cell): class
[all …]
/third_party/mindspore/tests/st/control/inner/
Dtest_030_for_in_if.py51 class GradNet(nn.Cell): class
53 super(GradNet, self).__init__()
64 net = GradNet(for_in_if_net)
102 class GradNet(nn.Cell): class
104 super(GradNet, self).__init__()
115 net = GradNet(for_in_if_net)
154 class GradNet(nn.Cell): class
156 super(GradNet, self).__init__()
167 net = GradNet(for_in_if_net)
204 class GradNet(nn.Cell): class
[all …]
Dtest_002_single_for.py47 class GradNet(nn.Cell): class
49 super(GradNet, self).__init__()
65 net = GradNet(for_net)
91 class GradNet(nn.Cell): class
93 super(GradNet, self).__init__()
106 net = GradNet(for_net)
145 class GradNet(nn.Cell): class
147 super(GradNet, self).__init__()
159 net = GradNet(single_for_net)
191 class GradNet(nn.Cell): class
[all …]
Dtest_331_for_after_for_in_while.py64 class GradNet(nn.Cell): class
66 super(GradNet, self).__init__()
78 net = GradNet(for_after_for_in_while_net)
87 net = GradNet(for_after_for_in_while_net)
122 class GradNet(nn.Cell): class
124 super(GradNet, self).__init__()
136 net = GradNet(for_after_for_in_while_net)
145 net = GradNet(for_after_for_in_while_net)
Dtest_310_for_after_if_in_if.py52 class GradNet(nn.Cell): class
54 super(GradNet, self).__init__()
65 net = GradNet(for_after_if_in_if_net)
102 class GradNet(nn.Cell): class
104 super(GradNet, self).__init__()
115 net = GradNet(for_after_if_in_if_net)
Dtest_031_for_in_while.py52 class GradNet(nn.Cell): class
54 super(GradNet, self).__init__()
65 backward_net = GradNet(for_in_while_net)
99 class GradNet(nn.Cell): class
101 super(GradNet, self).__init__()
112 net = GradNet(for_in_while_net)
Dtest_032_for_in_for.py57 class GradNet(nn.Cell): class
59 super(GradNet, self).__init__()
70 net = GradNet(for_in_for_net)
105 class GradNet(nn.Cell): class
107 super(GradNet, self).__init__()
118 net = GradNet(for_in_for_net)
Dtest_131_if_after_for_in_while.py44 class GradNet(nn.Cell): class
46 super(GradNet, self).__init__()
57 net = GradNet(if_after_for_in_while_net)
66 net = GradNet(if_after_for_in_while_net)
Dtest_102_if_after_for.py58 class GradNet(nn.Cell): class
60 super(GradNet, self).__init__()
71 net = GradNet(if_after_for_net)
112 class GradNet(nn.Cell): class
114 super(GradNet, self).__init__()
125 net = GradNet(if_after_for_net)
Dtest_302_for_after_for.py60 class GradNet(nn.Cell): class
62 super(GradNet, self).__init__()
73 net = GradNet(for_after_for_net)
113 class GradNet(nn.Cell): class
115 super(GradNet, self).__init__()
126 net = GradNet(for_after_for_net)
Dtest_330_for_after_for_in_if.py50 class GradNet(nn.Cell): class
52 super(GradNet, self).__init__()
63 net = GradNet(for_after_for_in_if_net)
72 net = GradNet(for_after_for_in_if_net)
Dtest_332_for_after_for_in_for.py64 class GradNet(nn.Cell): class
66 super(GradNet, self).__init__()
78 net = GradNet(for_after_for_in_for_net)
121 class GradNet(nn.Cell): class
123 super(GradNet, self).__init__()
135 net = GradNet(for_after_for_in_for_net)
Dtest_320_for_after_while_in_if.py72 class GradNet(nn.Cell): class
74 super(GradNet, self).__init__()
86 net = GradNet(for_after_while_in_if_net)
131 class GradNet(nn.Cell): class
133 super(GradNet, self).__init__()
145 net = GradNet(for_after_while_in_if_net)
Dtest_322_for_after_while_in_for.py73 class GradNet(nn.Cell): class
75 super(GradNet, self).__init__()
87 net = GradNet(for_after_while_in_for_net)
133 class GradNet(nn.Cell): class
135 super(GradNet, self).__init__()
147 net = GradNet(for_after_while_in_for_net)
/third_party/mindspore/tests/st/control/
Dtest_cont_grad.py46 class GradNet(nn.Cell): class
48 super(GradNet, self).__init__()
60 net = GradNet(while_net)
83 class GradNet(nn.Cell): class
85 super(GradNet, self).__init__()
93 net = GradNet(while_net)
120 class GradNet(nn.Cell): class
122 super(GradNet, self).__init__()
130 net = GradNet(while_net)
229 class GradNet(nn.Cell): class
[all …]
Dtest_for_to_while.py44 class GradNet(nn.Cell): class
46 super(GradNet, self).__init__()
60 net = GradNet(for_net)
67 net = GradNet(for_net)
/third_party/mindspore/tests/ut/python/pipeline/infer/
Dtest_auto_monad.py63 class GradNet(nn.Cell): class
65 super(GradNet, self).__init__()
73 net = GradNet(assign_net)
92 class GradNet(nn.Cell): class
94 super(GradNet, self).__init__()
102 net = GradNet(assign_net)
153 class GradNet(nn.Cell): class
155 super(GradNet, self).__init__()
163 net = GradNet(user_defined_net)
188 class GradNet(nn.Cell): class
[all …]
/third_party/mindspore/tests/ut/python/pipeline/parse/
Dtest_sequence_assign.py190 class GradNet(nn.Cell): class
192 super(GradNet, self).__init__()
200 grad_net = GradNet(net)
218 class GradNet(nn.Cell): class
220 super(GradNet, self).__init__()
228 grad_net = GradNet(net)
/third_party/mindspore/tests/st/ops/gpu/
Dtest_adaptive_avg_pool2d_op.py40 class GradNet(nn.Cell): class
42 super(GradNet, self).__init__()
108 grad_net = GradNet()
140 grad_net = GradNet()
172 grad_net = GradNet()
/third_party/mindspore/tests/st/ops/cpu/
Dtest_sparse_to_dense_op.py36 class GradNet(nn.Cell): class
38 super(GradNet, self).__init__()
63 grad_net = GradNet(forward_net)
90 grad_net = GradNet(forward_net)
106 grad_net = GradNet(forward_net)
Dtest_sparse_tensor_dense_matmul_op.py36 class GradNet(nn.Cell): class
38 super(GradNet, self).__init__()
70 grad_net = GradNet(sparse_dense_matmul_net)
100 grad_net = GradNet(sparse_dense_matmul_net)
130 grad_net = GradNet(sparse_dense_matmul_net)
160 grad_net = GradNet(sparse_dense_matmul_net)

123