Searched refs:cxx_pytree (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_pytree.py | 31 cxx_pytree = py_pytree variable
33 import torch.utils._cxx_pytree as cxx_pytree namespace
48 self.assertEqual(public_apis, cxx_pytree.__all__)
51 cxx_api = getattr(cxx_pytree, name)
133 subtest(cxx_pytree, name="cxx"),
172 subtest(cxx_pytree, name="cxx"),
203 (cxx_pytree, lambda tup: cxx_pytree.tree_structure((0,) * len(tup))),
238 (cxx_pytree, lambda lst: cxx_pytree.tree_structure([0] * len(lst))),
275 cxx_pytree,
276 lambda dct: cxx_pytree.tree_structure(dict.fromkeys(dct, 0)),
[all …]
|