Searched defs:pytree_impl (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_pytree.py | 136 def test_register_pytree_node(self, pytree_impl): argument
175 def test_flatten_unflatten_leaf(self, pytree_impl): argument
208 def test_flatten_unflatten_tuple(self, pytree_impl, gen_expected_fn): argument
243 def test_flatten_unflatten_list(self, pytree_impl, gen_expected_fn): argument
282 def test_flatten_unflatten_dict(self, pytree_impl, gen_expected_fn): argument
325 def test_flatten_unflatten_ordereddict(self, pytree_impl, gen_expected_fn): argument
369 def test_flatten_unflatten_defaultdict(self, pytree_impl, gen_expected_fn): argument
411 def test_flatten_unflatten_deque(self, pytree_impl, gen_expected_fn): argument
435 def test_flatten_unflatten_namedtuple(self, pytree_impl): argument
471 def test_flatten_unflatten_return_types(self, pytree_impl, op): argument
[all …]
|