Searched defs:scatter_reduce (Results 1 – 3 of 3) sorted by relevance
121 def scatter_reduce( function
252 TORCH_META_FUNC2(scatter_reduce, two) in TORCH_META_FUNC2() argument
3413 def scatter_reduce(x, dim: int, index, src, reduction_type, **kwargs): function