Searched refs:scatter_reduce_two_fn (Results 1 – 1 of 1) sorted by relevance
26 using scatter_reduce_two_fn = void(*)(const Tensor& self, const int64_t dim, const Tensor& index, variable37 DECLARE_DISPATCH(scatter_reduce_two_fn, scatter_reduce_two_stub);