Home
last modified time | relevance | path

Searched refs:normalize_axis_tuple (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_numpy/
H A D_util.py56 def normalize_axis_tuple(axis, ndim, argname=None, allow_duplicate=False): function
114 axis = normalize_axis_tuple(axis, out_ndim)
H A D_reductions_impl.py38 axis = _util.normalize_axis_tuple(axis, a.ndim)
406 axis = _util.normalize_axis_tuple(axis, a.ndim)
H A D_funcs_impl.py690 axis = _util.normalize_axis_tuple(axis, m.ndim)
703 axes = _util.normalize_axis_tuple(axes, m.ndim)
1417 source = _util.normalize_axis_tuple(source, a.ndim, "source")
1418 destination = _util.normalize_axis_tuple(destination, a.ndim, "destination")
1457 axis = _util.normalize_axis_tuple(axis, a.ndim, allow_duplicate=True)
1558 axes = _util.normalize_axis_tuple(axis, N)
/aosp_15_r20/external/pytorch/test/torch_np/
H A Dtest_reductions.py476 axis_norm = _util.normalize_axis_tuple(axis, d.ndim)
/aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/lib/
H A Dtest_function_base.py67 from numpy.core.numeric import normalize_axis_tuple
99 from torch._numpy._util import normalize_axis_tuple
3288 axis_norm = normalize_axis_tuple(axis, d.ndim)
3854 axis_norm = normalize_axis_tuple(axis, d.ndim)