| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_gathernd_further.py | 78 class Net3(Cell): class
190 net = Net3(w1_shape, indices_shape, strategy1, strategy2, strategy3)
202 net = Net3(w1_shape, indices_shape, strategy1, strategy2, strategy3)
214 net = Net3(w1_shape, indices_shape, strategy1, strategy2, strategy3)
298 net = Net3(w1_shape, indices_shape, strategy1, strategy2, strategy3)
310 net = Net3(w1_shape, indices_shape, strategy1, strategy2, strategy3)
322 net = Net3(w1_shape, indices_shape, strategy1, strategy2, strategy3)
|
| D | test_layer_norm_further.py | 74 class Net3(Cell): class
184 net = Net3(1, 1, _w, [64, 32, 16], strategy1, strategy2, strategy3)
234 net = Net3(2, 1, _w, [64, 32, 16], strategy1, strategy2, strategy3)
284 net = Net3(3, 1, _w, [64, 32, 16], strategy1, strategy2, strategy3)
334 net = Net3(3, 1, _w, [64, 32, 16], strategy1, strategy2, strategy3)
408 net = Net3(3, 1, _w, [64, 32, 16], strategy1, strategy2, strategy3)
|
| D | test_resizebilinear.py | 59 class Net3(Cell): class
65 super(Net3, self).__init__()
125 net = Net3(_w1, out_channel=8, kernel_size=2, pad_mode="same", stride=1)
|
| D | test_tile.py | 55 class Net3(Cell): class
125 net = Net3(_w1, strategy1, strategy2)
161 net = Net3(_w1)
|
| D | test_parameter_multi_users.py | 50 class Net3(Cell): class
115 net = Net3(_w, strategy1, strategy2)
|
| D | test_parallel_optimizer.py | 69 class Net3(nn.Cell): class
72 super(Net3, self).__init__()
127 …train_network = auto_parallel_compile_net("semi_auto_parallel", 32, Net3, ((4, 8), (8, 1)), ((4, 4…
|
| D | test_concat.py | 53 class Net3(Cell): class
155 net = Net3(w1, w2, w3)
|
| D | test_model_without_loss.py | 76 class Net3(Cell): class
189 net = Net3(w1, w2, w3)
|
| /third_party/mindspore/tests/st/probability/bijector/ |
| D | test_scalar_affine.py | 82 class Net3(nn.Cell): class
87 super(Net3, self).__init__()
94 backward_jacobian = Net3()
|
| D | test_exp.py | 87 class Net3(nn.Cell): class
92 super(Net3, self).__init__()
101 inverse_jacobian = Net3()
|
| D | test_invert.py | 83 class Net3(nn.Cell): class
88 super(Net3, self).__init__()
97 backward_jacobian = Net3()
|
| D | test_softplus.py | 82 class Net3(nn.Cell): class
87 super(Net3, self).__init__()
94 backward_jacobian = Net3()
|
| D | test_power_transform.py | 90 class Net3(nn.Cell): class
95 super(Net3, self).__init__()
105 inverse_jacobian = Net3(power=power)
|
| D | test_gumbel_cdf.py | 89 class Net3(nn.Cell): class
94 super(Net3, self).__init__()
103 backward_jacobian = Net3(loc, scale)
|
| /third_party/mindspore/tests/ut/python/pipeline/parse/ |
| D | test_celllist.py | 31 class Net3(Cell): class
46 net = Net3()
|
| D | test_dictionary.py | 52 class Net3(Cell): class
85 net = Net3()
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_recompute.py | 54 class Net3(Cell): class
56 super(Net3, self).__init__()
169 test_basic2(Net3)
207 test_basic2(Net3)
|
| /third_party/mindspore/tests/ut/python/optimizer/ |
| D | test_debug_location.py | 132 class Net3(nn.Cell): class
146 net = Net3()
|