| /third_party/mindspore/mindspore-src/source/tests/st/pi_jit/operation/ |
| D | test_pow.py | 254 def __init__(self, input_np): argument
288 def __init__(self, input_np): argument
321 def __init__(self, input_np): argument
|
| /third_party/mindspore/mindspore-src/source/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/mindspore-src/source/tests/st/ops/ |
| D | test_ops_index_select_ext.py | 39 def generate_expect_forward_output(input_np, axis, index_np): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/pi_jit/dynamic_shape_operations/ |
| D | abs.py | 24 def __init__(self, input_np, indices_np, dtype=np.float32): argument
|
| /third_party/mindspore/mindspore-src/source/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/mindspore-src/source/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/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_incre_flash_attention.py | 138 def change_data_bf16(input_np): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/pynative/ |
| D | test_tensor_setitem.py | 28 def setup_testcase(input_np, case_fn): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/ |
| D | test_dynamic_getitem.py | 59 def __init__(self, ms_net, np_net, input_np, input_dyn): argument
82 def __init__(self, ms_net, np_net, input_np, axis_np): argument
|
| D | test_dynamic_setitem.py | 106 def __init__(self, ms_net, np_net, input_np, axis_np): argument
|