tensorplay.profiler.schedule

tensorplay.profiler.schedule(*, wait: int, warmup: int, active: int, repeat: int = 0, skip_first: int = 0, skip_first_wait: int = 0)[source]

Build a callable that maps step numbers to profiler actions.