Home
last modified time | relevance | path

Searched defs:split_size_or_sections (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/functorch/dim/
H A Dreference.py599 def split(self, split_size_or_sections, dim=0): argument
/aosp_15_r20/external/pytorch/functorch/csrc/dim/
H A Ddim.cpp2760 mpy::handle self, split_size_or_sections, dim; in py_split() local