Searched refs:diagonal_scatter_batch_rule (Results 1 – 1 of 1) sorted by relevance
905 std::tuple<Tensor, std::optional<int64_t>> diagonal_scatter_batch_rule( in diagonal_scatter_batch_rule() function1246 VMAP_SUPPORT(diagonal_scatter, diagonal_scatter_batch_rule); in TORCH_LIBRARY_IMPL()