Searched refs:to_subclass (Results 1 – 4 of 4) sorted by relevance
2599 def to_subclass(t: torch.Tensor): function2604 args_subclass = pytree.tree_map_only(torch.Tensor, to_subclass, args)2605 kwargs_subclass = pytree.tree_map_only(torch.Tensor, to_subclass, kwargs)
416 def to_subclass(dt): function447 dt = to_subclass(example)
1162 def to_subclass(t, cls): function
4219 def to_subclass(x): function4223 tmp_subclass = to_subclass(x)