Searched refs:ModuleTracker (Results 1 – 4 of 4) sorted by relevance
9 from torch.utils.module_tracker import ModuleTracker41 with ModuleTracker() as tracker:92 with ModuleTracker() as tracker:97 with ModuleTracker() as tracker:104 with ModuleTracker() as tracker:
8 .. autoclass:: torch.utils.module_tracker.ModuleTracker
21 class ModuleTracker: class
5 from .module_tracker import ModuleTracker628 self.mod_tracker = ModuleTracker()