Searched refs:one_arg_scalar_funcs (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/torch_np/ |
| H A D | test_basic.py | 218 one_arg_scalar_funcs = [(w.size, _np.size), (w.shape, _np.shape), (w.ndim, _np.ndim)] variable
225 @parametrize("func, np_func", one_arg_scalar_funcs)
234 @parametrize("func, np_func", one_arg_scalar_funcs)
243 @parametrize("func, np_func", one_arg_scalar_funcs)
|