Home
last modified time | relevance | path

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

/external/pytorch/torch/_inductor/
Dlowering.py5063 def _validate_reduction_axis(x, axis): function
5085 axis = set(_validate_reduction_axis(x, axis))
5169 axis = _validate_reduction_axis(x, axis)
5186 axis = _validate_reduction_axis(x, axis)
5209 axis = _validate_reduction_axis(x, axis)
5244 axis = _validate_reduction_axis(x, axis)