Searched refs:torch_pytree (Results 1 – 2 of 2) sorted by relevance
10 from torch.utils import _pytree as torch_pytree unknown278 dynamic_shapes = torch_pytree.tree_map(dynamify_inp, inp)
84 from torch.utils import _pytree as torch_pytree unknown