| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_sparse_add_ops.py | 35 def sparse_add_same_indices(value_type, thresh_type, np_type, thresh_value): argument
54 def sparse_add_left_same_indices(value_type, thresh_type, np_type, thresh_value): argument
73 def sparse_add_right_same_indices(value_type, thresh_type, np_type, thresh_value): argument
92 def sparse_add_no_same_indices(value_type, thresh_type, np_type, thresh_value): argument
|
| D | test_batchnorm_grad_grad_op.py | 44 def get_nhwc_inputs(np_type): argument
63 def get_nchw_inputs(np_type): argument
82 def get_nc_inputs(np_type): argument
|
| D | test_dynamic_shape_op.py | 36 def dynamic_shape(np_type): argument
|
| D | test_hswish_op.py | 59 def generate_test_cases(np_type, mode, loss): argument
|
| D | test_argminwithvalue_op.py | 106 def argminwithvalue_tensor(context_mode, np_type): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_sparse_add_op.py | 40 def sparse_add_same_indices(value_type, thresh_type, np_type, thresh_value): argument
61 def sparse_add_left_same_indices(value_type, thresh_type, np_type, argument
83 def sparse_add_right_same_indices(value_type, thresh_type, np_type, argument
105 def sparse_add_no_same_indices(value_type, thresh_type, np_type, thresh_value): argument
|
| D | test_nllloss_op.py | 51 np_type = np.float32 variable
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_sqrt.py | 44 def sqrt_forward_test(ms_type, np_type): argument
57 def sqrt_backward_test(ms_type, np_type): argument
|
| D | test_assign_add.py | 40 def assign_add_forward_test(ms_type, np_type): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/dataset/ |
| D | test_to_number_op.py | 106 def out_of_bounds_error_message_check(dataset, np_type, value_to_cast): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/ |
| D | test_nllloss_dyn.py | 50 np_type = np.float32 variable
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_hardswish_op.py | 92 def generate_test_cases(np_type, mode): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_flatten.py | 137 def flatten_net_dynamic(np_type, ms_type): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/np_dtype/ |
| D | np_dtypes.cc | 1153 bool RegisterNpTypeCast(int np_type, bool scalar_castable) { in RegisterNpTypeCast()
|