Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/functorch/csrc/dim/
H A Ddim.cpp947 Unflatten tree_flatten(Arena& A, mpy::handle agg, Slice<mpy::handle>& flat_elements) { in tree_flatten()
1001 UnflattenVectorArgs tree_flatten(Arena& A, mpy::vector_args args, Slice<mpy::handle>& flat_elements in tree_flatten()