You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
545 B
23 lines
545 B
# 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 specified 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()
|