Searched refs:tensor_split_sections_batching_rule (Results 1 – 1 of 1) sorted by relevance
219 std::vector<Tensor> tensor_split_sections_batching_rule(const Tensor& self, int64_t sections, int64… in tensor_split_sections_batching_rule() function1103 m.impl("tensor_split.sections", tensor_split_sections_batching_rule); in TORCH_LIBRARY_IMPL()