(fn)
| 100 | # Function decorator that calls torch.autograd.profiler.record_function(). |
| 101 | |
| 102 | def profiled_function(fn): |
| 103 | def decorator(*args, **kwargs): |
| 104 | with torch.autograd.profiler.record_function(fn.__name__): |
| 105 | return fn(*args, **kwargs) |
| 106 | decorator.__name__ = fn.__name__ |
| 107 | return decorator |
| 108 | |
| 109 | #---------------------------------------------------------------------------- |
| 110 | # Sampler for torch.utils.data.DataLoader that loops over the dataset |
nothing calls this directly
no outgoing calls
no test coverage detected