pytorch_pfn_extras.profiler.record_function#

pytorch_pfn_extras.profiler.record_function(tag, use_cuda=False, enable=True, device='cpu', trace=False)#
Parameters:

  • tag (Optional[str]) –

  • use_cuda (bool) –

  • enable (bool) –

  • device (DeviceLike) –

  • trace (Union[Tracer, bool]) –

Return type:

Callable[[Callable[[…], _T]], Callable[[…], _T]]