Searched defs:input_np (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_tanh.py | 36 input_np = np.random.randn(*input_shape).astype(np.float32) variable
|
| D | test_tanh_grad.py | 36 input_np = np.random.randn(*input_shape).astype(np.float32) variable
|
| /third_party/mindspore/tests/ut/python/exec/ |
| D | test_train_with_lars.py | 79 def me_train_tensor(net, input_np, label_np, epoch_size=2): argument
|
| D | test_train.py | 32 def me_train_tensor(net, input_np, label_np, epoch_size=2): argument
|
| /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): argument
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_tensor_setitem.py | 28 def setup_testcase(input_np, case_fn): argument
|