Searched refs:tensor_split_sections_symint (Results 1 – 3 of 3) sorted by relevance
295 m.impl("tensor_split.sections", native::tensor_split_sections_symint); in TORCH_LIBRARY_IMPL()
877 pnode->Output(0) = at::native::tensor_split_sections_symint(a, b, c); in __anon75e5f0515402()
959 std::vector<Tensor> tensor_split_sections_symint(const Tensor& self, c10::SymInt sym_sections, int6… in tensor_split_sections_symint() function