Searched refs:scatter_src_batch_rule (Results 1 – 1 of 1) sorted by relevance
754 std::tuple<Tensor, std::optional<int64_t>> scatter_src_batch_rule( in scatter_src_batch_rule() function1249 VMAP_SUPPORT2(scatter, src, scatter_src_batch_rule); in TORCH_LIBRARY_IMPL()