Searched refs:_replace_tuple_with_list (Results 1 – 1 of 1) sorted by relevance
139 def _replace_tuple_with_list(spec: pytree.TreeSpec) -> pytree.TreeSpec: function142 _type, spec.context, list(map(_replace_tuple_with_list, spec.children_specs))170 lambda: _replace_tuple_with_list(spec1) == _replace_tuple_with_list(spec2),