| import torch | |
| add_stat_value = torch.ops.prim.AddStatValue | |
| set_logger = torch._C._logging_set_logger | |
| LockingLogger = torch._C.LockingLogger | |
| AggregationType = torch._C.AggregationType | |
| NoopLogger = torch._C.NoopLogger | |
| time_point = torch.ops.prim.TimePoint | |