| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_broadcast_to.py | 69 train_net = TrainOneStepCell(net, optimizer)
70 train_net.set_auto_parallel()
71 train_net.set_train()
72 _cell_graph_executor.compile(train_net, _x1)
79 train_net = TrainOneStepCell(net, optimizer)
80 train_net.set_auto_parallel()
81 train_net.set_train()
82 _cell_graph_executor.compile(train_net, _x1, _x2)
|
| D | test_pack.py | 141 train_net = TrainOneStepCell(net, optimizer)
142 train_net.set_auto_parallel()
143 train_net.set_train()
144 _cell_graph_executor.compile(train_net, _x)
151 train_net = TrainOneStepCell(net, optimizer)
152 train_net.set_auto_parallel()
153 train_net.set_train()
154 _cell_graph_executor.compile(train_net, _x1)
161 train_net = TrainOneStepCell(net, optimizer)
162 train_net.set_auto_parallel()
[all …]
|
| D | test_batchnorm.py | 47 train_net = TrainOneStepCell(net, optimizer)
48 train_net.set_auto_parallel()
49 train_net.set_train()
50 _cell_graph_executor.compile(train_net, _x, _b)
97 train_net = TrainOneStepCell(net, optimizer)
98 train_net.set_auto_parallel()
99 train_net.set_train()
100 _cell_graph_executor.compile(train_net, _x1, _b1)
|
| D | test_print.py | 49 train_net = TrainOneStepCell(net, optimizer)
50 train_net.set_auto_parallel()
51 train_net.set_train()
52 _cell_graph_executor.compile(train_net, _x, _b)
102 train_net = TrainOneStepCell(net, optimizer)
103 train_net.set_auto_parallel()
104 train_net.set_train()
105 _cell_graph_executor.compile(train_net, _x1, _b1)
|
| D | test_dynamic_shape.py | 81 train_net = TrainOneStepCell(net, optimizer)
82 train_net.set_auto_parallel()
83 train_net.set_train()
84 _cell_graph_executor.compile(train_net, x)
115 train_net = TrainOneStepCell(net, optimizer)
116 train_net.set_auto_parallel()
117 train_net.set_train()
118 _cell_graph_executor.compile(train_net, x)
|
| D | test_split.py | 80 train_net = TrainOneStepCell(net, optimizer)
81 train_net.set_auto_parallel()
82 train_net.set_train()
83 _cell_graph_executor.compile(train_net, _x)
90 train_net = TrainOneStepCell(net, optimizer)
91 train_net.set_auto_parallel()
92 train_net.set_train()
93 _cell_graph_executor.compile(train_net, _x1)
|
| D | test_loss_and_optimizer.py | 73 train_net = TrainOneStepCell(net_with_loss, optimizer)
75 compile_net(train_net, x, b)
107 train_net = TrainOneStepCell(net_with_loss, optimizer)
109 compile_net(train_net, x, b)
142 train_net = TrainOneStepCell(net_with_loss, optimizer)
144 compile_net(train_net, x, b)
178 train_net = TrainOneStepCell(net_with_loss, optimizer)
180 compile_net(train_net, x, b)
213 train_net = TrainOneStepCell(net_with_loss, optimizer)
215 compile_net(train_net, x, b)
|
| D | test_reshape_optimized.py | 44 train_net = TrainOneStepCell(net, optimizer)
45 train_net.set_auto_parallel()
46 train_net.set_train()
47 _cell_graph_executor.compile(train_net, _x, _b)
|
| D | test_auto_parallel_activation.py | 44 train_net = TrainOneStepCell(net, optimizer)
45 train_net.set_auto_parallel()
46 train_net.set_train()
47 _cell_graph_executor.compile(train_net, _x, _b)
|
| D | test_sparse_feature_bprop.py | 97 train_net = TrainOneStepCell(net, optimizer)
98 train_net.set_train()
99 _cell_graph_executor.compile(train_net, _x, _b)
129 train_net = TrainOneStepCell(net, optimizer)
130 train_net.set_train()
131 _cell_graph_executor.compile(train_net, _x, _b)
|
| D | test_batchparallel_replace_shape.py | 47 train_net = TrainOneStepCell(net, optimizer)
48 train_net.set_auto_parallel()
49 train_net.set_train()
50 _cell_graph_executor.compile(train_net, _x, _b)
|
| D | test_reshape_skip_redistribution.py | 47 train_net = TrainOneStepCell(net, optimizer)
48 train_net.set_auto_parallel()
49 train_net.set_train()
50 _cell_graph_executor.compile(train_net, _x, _b)
|
| D | test_auto_parallel_shard_propagation2.py | 46 train_net = TrainOneStepCell(net, optimizer)
47 train_net.set_auto_parallel()
48 train_net.set_train()
49 _cell_graph_executor.compile(train_net, _x, _b)
|
| D | test_reluv2.py | 45 train_net = TrainOneStepCell(net, optimizer)
46 train_net.set_auto_parallel()
47 train_net.set_train()
48 _cell_graph_executor.compile(train_net, _x)
|
| D | test_gatherd.py | 45 train_net = TrainOneStepCell(net, optimizer)
46 train_net.set_auto_parallel()
47 train_net.set_train()
48 _cell_graph_executor.compile(train_net, _x, _b)
|
| D | test_sigmoid_cross_entropy_with_logits.py | 44 train_net = TrainOneStepCell(net, optimizer)
45 train_net.set_auto_parallel()
46 train_net.set_train()
47 _cell_graph_executor.compile(train_net, _x, _b)
|
| D | test_auto_parallel_multi_graph.py | 59 train_net = TrainOneStepCell(net, optimizer)
60 train_net.set_auto_parallel()
61 train_net.set_train()
62 _cell_graph_executor.compile(train_net, inputs_, label_)
|
| D | test_one_weight_parameter.py | 75 train_net = OneStepCell(net_with_loss)
77 train_net.set_auto_parallel()
79 train_net.set_train()
80 _cell_graph_executor.compile(train_net, x, b)
|
| D | test_adafactor.py | 53 train_net = TrainOneStepCell(net, optimizer)
54 train_net.set_auto_parallel()
55 train_net.set_train()
56 _cell_graph_executor.compile(train_net, _x, _b)
|
| D | test_onehot_2dim.py | 50 train_net = TrainOneStepCell(net, optimizer)
51 train_net.set_auto_parallel()
52 train_net.set_train()
53 _cell_graph_executor.compile(train_net, _x)
|
| D | test_neg.py | 44 train_net = TrainOneStepCell(net, optimizer)
45 train_net.set_auto_parallel()
46 train_net.set_train()
47 _cell_graph_executor.compile(train_net, _x, _b)
|
| D | test_auto_parallel_shard_propagation3.py | 48 train_net = TrainOneStepCell(net, optimizer)
49 train_net.set_auto_parallel()
50 train_net.set_train()
51 _cell_graph_executor.compile(train_net, _x, _b)
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | train_and_test_multinpu_ci.py | 43 train_net = TrainStepWrap(loss_net)
46 return train_net, eval_net
89 train_net, eval_net = net_builder.get_net(config)
90 train_net.set_train()
93 model = Model(train_net, eval_network=eval_net, metrics={"auc": auc_metric})
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/ |
| D | train_and_test_multinpu_ci_data_parallel.py | 40 train_net = TrainStepWrap(loss_net)
42 return train_net, eval_net
87 train_net, eval_net = net_builder.get_net(config)
88 train_net.set_train()
91 model = Model(train_net, eval_network=eval_net, metrics={"auc": auc_metric})
|
| /third_party/mindspore/mindspore/lite/examples/train_lenet/model/ |
| D | train_utils.py | 36 train_net = nn.TrainOneStepCell(loss_net, optimizer)
38 …train_net = amp.build_train_network(net, optimizer, loss_fn, level="O2", loss_scale_manager=loss_s…
39 return train_net
|