Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/functorch/dim/
H A Dtree_map.py13 def tree_map(fn, tree): function
/aosp_15_r20/external/executorch/extension/pytree/
H A D__init__.py31 tree_map as tree_map, unknown
H A Dpybindings.cpp298 static py::object tree_map(py::function& fn, py::handle x) { in tree_map() function
/aosp_15_r20/external/pytorch/torch/utils/
H A D_cxx_pytree.py426 def tree_map( function
H A D_pytree.py923 def tree_map( function
/aosp_15_r20/prebuilts/gcc/linux-x86/host/x86_64-w64-mingw32-4.8/lib/gcc/x86_64-w64-mingw32/4.8.3/plugin/include/
Dcgraph.h196 vec<ipa_replace_map_p, va_gc> *tree_map; member
Dtree.h6247 struct GTY(()) tree_map { struct
6248 struct tree_map_base base; argument
6253 #define tree_map_eq tree_map_base_eq argument
/aosp_15_r20/external/pytorch/functorch/csrc/dim/
H A Ddim.cpp1089 mpy::object tree_map(Arena& A, const std::function<mpy::handle(mpy::handle)>& fn, mpy::handle agg) { in tree_map() function