Searched refs:scatter_value_reduce_batch_rule (Results 1 – 1 of 1) sorted by relevance
782 std::tuple<Tensor, std::optional<int64_t>> scatter_value_reduce_batch_rule( in scatter_value_reduce_batch_rule() function1252 VMAP_SUPPORT2(scatter, value_reduce, scatter_value_reduce_batch_rule); in TORCH_LIBRARY_IMPL()