Searched refs:split_with_sizes_batching_rule (Results 1 – 2 of 2) sorted by relevance
249 std::vector<Tensor> split_with_sizes_batching_rule(const Tensor& self, SymIntArrayRef split_sizes, … in split_with_sizes_batching_rule() function738 m.impl("split_with_sizes", split_with_sizes_batching_rule); in TORCH_LIBRARY_IMPL()
440 std::vector<Tensor> split_with_sizes_batching_rule(const Tensor& self, IntArrayRef split_sizes, int… in split_with_sizes_batching_rule() function1124 m.impl("split.sizes", split_with_sizes_batching_rule); in TORCH_LIBRARY_IMPL()1125 m.impl("split_with_sizes", split_with_sizes_batching_rule); in TORCH_LIBRARY_IMPL()