Searched refs:treespec_pprint (Results 1 – 4 of 4) sorted by relevance
10 from torch.utils._pytree import tree_map_, treespec_pprint
42 treespec_pprint,
901 def treespec_pprint(treespec: TreeSpec) -> str: function
1463 def treespec_pprint(treespec: TreeSpec) -> str: function