23 lines
545 B
Python
23 lines
545 B
Python
# mypy: allow-untyped-defs
|
|
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)
|
|
|
|
|
|
def metrics_report():
|
|
"""Return the combined (lazy core and backend) metric report"""
|
|
return torch._C._lazy._metrics_report()
|