torch/_lazy/metrics.py

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)