Home
last modified time | relevance | path

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.py187 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.py47 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.py297 axis = _util.normalize_axis_index(axis, a.ndim)
314 axis = _util.normalize_axis_index(axis, a.ndim)