Searched refs:me_train_tensor (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/tests/ut/python/exec/ |
| D | test_train.py | 32 def me_train_tensor(net, input_np, label_np, epoch_size=2): function
80 me_train_tensor(Net(3, bias_init=bias_init), input_np, label_np)
112 me_train_tensor(net, input_np, label_np)
149 me_train_tensor(net, input_np, label_np)
179 me_train_tensor(net, input_np, label_np)
|
| D | test_train_with_lars.py | 79 def me_train_tensor(net, input_np, label_np, epoch_size=2): function
|
| /third_party/mindspore/tests/st/fusion/ |
| D | test_conv_bn1_fusion.py | 41 def me_train_tensor(net, input_np, label_np, epoch_size=2): function
84 me_train_tensor(net, input_np, label_np)
111 me_train_tensor(net, input_np, label_np)
136 me_train_tensor(net, input_np, label_np)
|