Searched refs:normalize_axis_index (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/torch/_numpy/ |
H A D | _funcs_impl.py | 187 axis = _util.normalize_axis_index(axis, result_ndim) 217 axis = _util.normalize_axis_index(axis, tensor.ndim) 881 axis = _util.normalize_axis_index(axis, a.ndim) 889 axis = _util.normalize_axis_index(axis, arr.ndim) 916 axis = _util.normalize_axis_index(axis, arr.ndim) 958 axis = _util.normalize_axis_index(axis, ar.ndim) 1029 axis1 = _util.normalize_axis_index(axis1, a.ndim) 1030 axis2 = _util.normalize_axis_index(axis2, a.ndim) 1223 axisa = _util.normalize_axis_index(axisa, a.ndim) 1224 axisb = _util.normalize_axis_index(axisb, b.ndim) [all …]
|
H A D | _util.py | 47 def normalize_axis_index(ax, ndim, argname=None): function 92 axis = tuple([normalize_axis_index(ax, ndim, argname) for ax in axis])
|
H A D | _reductions_impl.py | 297 axis = _util.normalize_axis_index(axis, a.ndim) 314 axis = _util.normalize_axis_index(axis, a.ndim)
|