12 lines
257 B
Python
12 lines
257 B
Python
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
|