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 | |