| import torch._C._lazy | |
| def reset(): | |
| """Resets all metric counters.""" | |
| torch._C._lazy._reset_metrics() | |
| def counter_names(): | |
| """Retrieves all the currently active counter names.""" | |
| return torch._C._lazy._counter_names() | |
| def counter_value(name: str): | |
| """Return the value of the counter with the speficied name""" | |
| return torch._C._lazy._counter_value(name) |