Searched refs:annotate_split_points (Results 1 – 1 of 1) sorted by relevance
1170 def annotate_split_points(mod: torch.nn.Module, spec: Dict[str, SplitPoint]): function1230 annotate_split_points(module, split_spec)