11 lines
256 B
Python
11 lines
256 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
|