profile

class tensorplay.profiler.profiler.profile(enabled=True, *, activities=None, schedule=None, on_trace_ready=None, record_shapes=False, profile_memory=False, with_stack=False, with_flops=False, with_modules=False, use_device=None, gpu_timing=False, gpu_trace=False, with_samples=False)[source]

Context manager that records dispatched operations and annotations.

step()[source]

Advance the configured step schedule.